[chore] tidy up val F1 score printing

gudgud96 · gudgud96 · commit b501f8375dc5 · 2024-06-10T18:38:38.000+08:00
diff --git a/tasks/mt3_base.py b/tasks/mt3_base.py
@@ -40,9 +40,8 @@ def on_validation_epoch_end(self):
                     ground_truth_midi_dir=self.eval_cfg.midi_dir,
                     verbose=False
                 )
-                print(scores)
 
-                self.log('val_f1_flat', scores['Onset F1'], on_step=False, on_epoch=True)
+                self.log('val_f1_flat', scores['Onset F1'], on_step=False, on_epoch=True, prog_bar=True)
                 self.log('val_f1_midi_class', scores['Onset + program F1 (midi_class)'], on_step=False, on_epoch=True)
                 self.log('val_f1_full', scores['Onset + program F1 (full)'], on_step=False, on_epoch=True)
     

Original file line number	Diff line number	Diff line change
`@@ -40,9 +40,8 @@ def on_validation_epoch_end(self):`
`40`	`40`	`ground_truth_midi_dir=self.eval_cfg.midi_dir,`
`41`	`41`	`verbose=False`
`42`	`42`	`)`
`43`		`- print(scores)`
`44`	`43`
`45`		`- self.log('val_f1_flat', scores['Onset F1'], on_step=False, on_epoch=True)`
	`44`	`+ self.log('val_f1_flat', scores['Onset F1'], on_step=False, on_epoch=True, prog_bar=True)`
`46`	`45`	`self.log('val_f1_midi_class', scores['Onset + program F1 (midi_class)'], on_step=False, on_epoch=True)`
`47`	`46`	`self.log('val_f1_full', scores['Onset + program F1 (full)'], on_step=False, on_epoch=True)`
`48`	`47`