From f8bbab7b5c56f692bddd0d2ca63766da7be0929a Mon Sep 17 00:00:00 2001
From: eneren <engin.eren@cern.ch>
Date: Sun, 2 Oct 2022 15:05:06 +0000
Subject: [PATCH] another typo
---
pytorch_job_wganSingleGen_ncc.yaml | 6 +++---
wganSingleGen.py | 2 +-
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/pytorch_job_wganSingleGen_ncc.yaml b/pytorch_job_wganSingleGen_ncc.yaml
index d857eee..9e52b51 100644
--- a/pytorch_job_wganSingleGen_ncc.yaml
+++ b/pytorch_job_wganSingleGen_ncc.yaml
@@ -35,12 +35,12 @@ spec:
value: "1"
command: [sh, -c]
args:
- - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --lrGen_E_H 0.00001 --chpt --chpt_eph 50 --batch-size 64 --ncrit 4
+ - python -u wganSingleGen.py --backend nccl --epochs 150 --exp wganSingleGenV1 --lrGen_E_H 0.00001 --chpt --chpt_eph 50 --batch-size 64 --ncrit 4
resources:
limits:
nvidia.com/gpu: 1
Worker:
- replicas: 3
+ replicas: 4
restartPolicy: OnFailure
template:
metadata:
@@ -70,7 +70,7 @@ spec:
value: "1"
command: [sh, -c]
args:
- - python -u wganSingleGen.py --backend nccl --epochs 50 --exp wganSingleGenV1 --lrGen_E_H 0.00001 --chpt --chpt_eph 50 --batch-size 64 --ncrit 4
+ - python -u wganSingleGen.py --backend nccl --epochs 150 --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 03840f9..8509436 100644
--- a/wganSingleGen.py
+++ b/wganSingleGen.py
@@ -299,7 +299,7 @@ def run(args):
optimizerD_E_H.load_state_dict(critic_E_H_checkpoint['optimizer_state_dict'])
Gen_E_H.load_state_dict(gen_E_H_checkpoint['model_state_dict'])
- optimizerG_E_H.load_state_dict(gen_H_E_checkpoint['optimizer_state_dict'])
+ optimizerG_E_H.load_state_dict(gen_E_H_checkpoint['optimizer_state_dict'])
eph = gen_E_H_checkpoint['epoch']
--
GitLab