Skip to content
Snippets Groups Projects
Commit 93599179 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

small fix

parent 9ae5047d
No related branches found
No related tags found
1 merge request!22Cross validation
Pipeline #27517 passed
......@@ -341,6 +341,6 @@ class MCCNNTrainer(object):
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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment