What does it look like when you instantiate the output_model
object?
Hi @<1592326527742119936:profile|ThoughtfulSeahorse27> , this indeed seems like a bug - can you please open a GitHub issue? 🙂
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'],)