From 30a9b017a3a9892c0e4fdcd221f0a0dbd1dfaa93 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Fri, 22 Jul 2022 11:17:29 +0200
Subject: [PATCH] forgot ECAL critic
---
wgan_ECAL_HCAL_2crit.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/wgan_ECAL_HCAL_2crit.py b/wgan_ECAL_HCAL_2crit.py
index 11113d4..57f9465 100644
--- a/wgan_ECAL_HCAL_2crit.py
+++ b/wgan_ECAL_HCAL_2crit.py
@@ -443,8 +443,14 @@ def run(args):
else:
eph = 0
gen_E_checkpoint = torch.load("/eos/user/e/eneren/experiments/wganv1_generator_694.pt", map_location=torch.device('cuda'))
+ critic_E_checkpoint = torch.load("/eos/user/e/eneren/experiments/wganv1_critic_694.pt", map_location=torch.device('cuda'))
+
mGenE.load_state_dict(gen_E_checkpoint['model_state_dict'])
optimizerG_E.load_state_dict(gen_E_checkpoint['optimizer_state_dict'])
+
+ mCritE.load_state_dict(critic_E_checkpoint['model_state_dict'])
+ optimizerD_E.load_state_dict(critic_E_checkpoint['optimizer_state_dict'])
+
print ("init models")
--
GitLab