changed path

Author: paulaharder

main.py

@@ -8,7 +8,7 @@
 def add_arguments():
     parser = argparse.ArgumentParser()
     parser.add_argument("--dataset", default="era5_twc", help="choose a data set to use")
-    parser.add_argument("--model", default="resnet2")
+    parser.add_argument("--model", default="cnn")
     parser.add_argument("--model_id", default="test")
     parser.add_argument("--upsampling_factor", default=4, type=int)
     parser.add_argument("--constraints", default="none") 

training.py

@@ -239,7 +239,7 @@ def save_dict(dictionary, args):
         w.writerow([key, val])
 
 def load_weights(model, model_id):
-    PATH = '/home/harder/constrained-downscaling/models/'+model_id+'.pth'
+    PATH = './models/'+model_id+'.pth'
     checkpoint = torch.load(PATH) # ie, model_best.pth.tar
     model.load_state_dict(checkpoint['state_dict'])
     model.to('cuda')