rev

cvlab-yonsei · Feb 4, 2021 · 39bea37 · 39bea37
1 parent 74d5809
commit 39bea37
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Train.py b/Train.py
@@ -152,10 +152,10 @@
     print(m_items)
     print('----------------------------------------')
 
-# print('Training is finished')
+print('Training is finished')
 # Save the model and the memory items
-    torch.save(model, os.path.join(log_dir, 'model_%02d.pth'%(epoch)))
-    torch.save(m_items, os.path.join(log_dir, 'keys_%02d.pt'%(epoch)))
+torch.save(model, os.path.join(log_dir, 'model.pth'))
+torch.save(m_items, os.path.join(log_dir, 'keys.pt'))
 
 sys.stdout = orig_stdout
 f.close()