diff --git a/pytorch_job_wganSingleGen_ncc.yaml b/pytorch_job_wganSingleGen_ncc.yaml
index e1c5db6b7d00832a3f84a35985121568ca41ec18..e0fe6f21f64b996d2cf0ebe529af5aad92e56e4b 100644
--- a/pytorch_job_wganSingleGen_ncc.yaml
+++ b/pytorch_job_wganSingleGen_ncc.yaml
@@ -35,7 +35,7 @@ spec:
                 value: "1"
               command: [sh, -c] 
               args:  
-                - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --batch-size 64 --ncrit 4  
+                - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --lrGen_E_H 0.00001 --chpt --chpt_eph 50 --batch-size 64 --ncrit 4  
               resources: 
                 limits:
                   nvidia.com/gpu: 1
@@ -70,7 +70,7 @@ spec:
                 value: "1"
               command: [sh, -c] 
               args:  
-                - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --batch-size 64 --ncrit 4
+                - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --lrGen_E_H 0.00001 --chpt --chpt_eph 50 --batch-size 64 --ncrit 4
               resources: 
                 limits:
                   nvidia.com/gpu: 1
diff --git a/wganSingleGen.py b/wganSingleGen.py
index 96e11b2e0dc1c932a6c88aa6c8ac5d6646936e30..4641cc6287d15fa749d431cbd4f62fad8a828846 100644
--- a/wganSingleGen.py
+++ b/wganSingleGen.py
@@ -292,7 +292,7 @@ def run(args):
     
     # load from checkpoint if desired
     if (args.chpt):
-        critic_E_H_checkpoint = torch.load(args.chpt_base + args.exp + "_criticE_"+ str(args.chpt_eph) + ".pt")
+        critic_E_H_checkpoint = torch.load(args.chpt_base + args.exp + "_critic_E_H_"+ str(args.chpt_eph) + ".pt")
         gen_E_H_checkpoint = torch.load(args.chpt_base + args.exp + "_generatorE_H_"+ str(args.chpt_eph) + ".pt")
 
         Crit_E_H.load_state_dict(critic_E_H_checkpoint['model_state_dict'])