Prepping for release submitted version

silkemaes · Mar 29, 2024 · 83f7ceb · 83f7ceb
1 parent c96427b
commit 83f7ceb
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 39 deletions.
diff --git a/figs/visualisation_integrated_training_64.png b/figs/visualisation_integrated_training_64.png
diff --git a/src/mace/CSE_0D/intregr_train.py b/src/mace/CSE_0D/intregr_train.py
@@ -143,13 +143,14 @@ def train(model, lr, data_loader, test_loader,nb_evol, path, end_epochs, DEVICE,
 
 
         ## save model every epoch temporarily
-        torch.save(model.state_dict(),path+'/nn/nn-1'+'.pt')
-        np.save(path+'/nb_epoch.npy',np.array([epoch+1]))
-        ## losses
-        trainpath = path+'/train'
-        testpath  = path+'/test'
-        trainloss.save(trainpath)
-        testloss.save(testpath)
+        if path != None:
+            torch.save(model.state_dict(),path+'/nn/nn-1'+'.pt')
+            np.save(path+'/nb_epoch.npy',np.array([epoch+1]))
+            ## losses
+            trainpath = path+'/train'
+            testpath  = path+'/test'
+            trainloss.save(trainpath)
+            testloss.save(testpath)
         ## plot
         # plotting.plot_loss(trainloss, testloss, log = log, show = show)
         # plt.savefig(path+'/loss.png')

diff --git a/train2.0_dev.ipynb b/train2.0_dev.ipynb