Hello all, I am trying to report a confusion matrix for my Output Model as follows: output_model.report_confusion_matrix(title = "Validation Confusion Matrix...
one year ago
Ok, thanks for trying
Reporting the confusion matrix AFTER updating the weights fixed the issue.
task = Task.init(project_name=params['project name'],
task_name= "-".join([params['task name'], params['net']['model'], 'lr', str(params['opt']['lr'])]),
auto_connect_frameworks={"tensorflow":False,"tensorboard": False,'pytorch': False,})
output_model = OutputModel(task=task,
framework="PyTorch",
config_dict = params,
tags = [params['project name']] + params['tags'],)...