From c0001547a6ab8f15b5212de3b2abbe4933a8ae1d Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Mon, 4 Jul 2022 15:57:43 +0200
Subject: [PATCH] correcting the unused variable
---
wganHCAL.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/wganHCAL.py b/wganHCAL.py
index a3dd6c1..262c458 100644
--- a/wganHCAL.py
+++ b/wganHCAL.py
@@ -68,10 +68,13 @@ def train(args, aD, aG, aGE, device, train_loader, optimizer_d, optimizer_g, epo
label = energy.to(device).float()
###
-
+ ## Generate Fake ECAL
+ zE = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz, 1, 1, 1))), requires_grad=False)
+ fake_ecal = aGE(zE, label.view(-1, 1, 1, 1, 1)).detach()
+ fake_ecal = fake_ecal.unsqueeze(1)
+
## Generate Fake HCAL
z = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz))), requires_grad=False)
-
fake_dataHCAL = aG(z, label, fake_ecal).detach() ## 48 x 30 x 30
## Critic fwd pass on Real
--
GitLab