Skip to content
Snippets Groups Projects

Cross validation

Merged Anjith GEORGE requested to merge cross_validation into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -341,6 +341,6 @@ class MCCNNTrainer(object):
@@ -341,6 +341,6 @@ class MCCNNTrainer(object):
self.network.load_state_dict(best_model_wts)
self.network.load_state_dict(best_model_wts)
# best epoch is named as best
# best epoch is named as 100
self.save_model(output_dir, epoch=100, iteration=0, losses=losses)
self.save_model(output_dir, epoch=100, iteration=0, losses=losses)
Loading