Models

finetune.py

@@ -126,6 +126,8 @@ if __name__ == '__main__':
     print('Saving model to', model_path)
     torch.save(model.module.state_dict(), model_path)
 
+    last_accuracy = 0.0
+    
     for epoch in range(args.epochs):
         # Train
         model.train()
@@ -161,7 +163,6 @@ if __name__ == '__main__':
         test_loader = test_dset
         model.eval()
 
-        last_accuracy = 0.0
 
         with torch.no_grad():
             print('*'*80)

helper.bash

@@ -1 +1,4 @@
-python finetune.py --data-location /l/users/u21010238/data/AiorNot --model-location /home/sara.pieri/Documents/model-soups/models --batch-size 56 --name finetune_cp_AiorNot_model_0 --checkpoint_path /home/sara.pieri/Documents/model-soups/models/model_0.pt
+python finetune.py --data-location /l/users/u21010238/data/AiorNot --model-location /home/sara.pieri/Documents/model-soups/models --batch-size 56 --name finetune_cp_AiorNot_model_1 --checkpoint_path /home/sara.pieri/Documents/model-soups/models/model_1.pt
+python finetune.py --data-location /l/users/u21010238/data/AiorNot --model-location /home/sara.pieri/Documents/model-soups/models --batch-size 56 --name finetune_cp_AiorNot_model_2 --checkpoint_path /home/sara.pieri/Documents/model-soups/models/model_2.pt
+python finetune.py --data-location /l/users/u21010238/data/AiorNot --model-location /home/sara.pieri/Documents/model-soups/models --batch-size 56 --name finetune_cp_AiorNot_model_3 --checkpoint_path /home/sara.pieri/Documents/model-soups/models/model_3.pt
+python finetune.py --data-location /l/users/u21010238/data/AiorNot --model-location /home/sara.pieri/Documents/model-soups/models --batch-size 56 --name finetune_cp_AiorNot_model_4 --checkpoint_path /home/sara.pieri/Documents/model-soups/models/model_4.pt

models/finetune_cp_AiorNot_model_0.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6fec067ce46bc2f297f4ed084a6bf73d8f873b0151ff896f7ee32c386b22b6b
+size 453864271

models/finetune_cp_AiorNot_model_1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3598bbb9fc6df5644d693253f3325e53971eadb842140a6e52d3dd5ae4bac23d
+size 453864271

models/finetune_cp_AiorNot_model_2.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f64db72147c161e3da2709ff331b47d6a996fb0280385e7155643110bfb92c44
+size 453864271

models/finetune_cp_AiorNot_model_3.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:027263b39345e7a358812b93b1133f45076e144ef01694479f080986f1d046b5
+size 453864271

models/finetune_cp_AiorNot_model_4.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ab7f44e7f45acdb836fd03dafc0b05a7754325c0bb9ebabc770faa590a09b8c
+size 453864271