Skip to content

Log Confusion Matrix to Comet Logger and fix for resume training

Summary

This MR introduces the following changes

  • Adds functionality to log the confusion matrix to comet logger, at the end of each epoch.

  • Don't recalculate class_weights for resumed training. They are already stored in the config file.

Conformity

Edited by Wei Sheng Lai

Merge request reports

Loading