diff --git a/image_ref/main.py b/image_ref/main.py index d579f81daa676d18208954b73bebc0c8720442f2..933794ad4ebbe513cac01d1246afd32868d794cd 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -203,7 +203,7 @@ def run_duo(args): else : wdb.log({ 'confidence matrix val': wdb.Image(args.base_out + '_confidence_matrix_val.png'), - 'confidence matrix test': wdb.Image(args.base_out + '_confidence_matrix_test.png'),}) + 'confusion matrix val' : wdb.Image(args.base_out+'_confusion_matrix_val.png'),}) wdb.finish()