MIC-DKFZ / medicaldetectiontoolkit

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Plot legend missing when using --resume_to_checkpoint

rtgunti opened this issue · comments

Hello! Thanks for the amazing repo!
Below is the training plot on a custom dataset when trained for 5 epochs.
train_pre
When I try to resume the training from the last checkpoint, the legend is missing from the updated training plot
train_resume