Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92)

This commit is contained in:
Glenn Jocher
2022-12-25 17:04:24 +01:00
committed by GitHub
parent 0298821467
commit 249dfbdc05
5 changed files with 17 additions and 24 deletions

View File

@ -214,8 +214,7 @@ class SegmentationTrainer(DetectionTrainer):
return dict(zip(keys, loss_items)) if loss_items is not None else keys
def progress_string(self):
return ('\n' + '%11s' * 8) % \
('Epoch', 'GPU_mem', *self.loss_names, 'Instances', 'Size')
return ('\n' + '%11s' * 8) % ('Epoch', 'GPU_mem', *self.loss_names, 'Instances', 'Size')
def plot_training_samples(self, batch, ni):
images = batch["img"]