From 32c31eab98472cbd722aaef115a759b91a2811b3 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Tue, 19 Jul 2022 13:56:12 +0200
Subject: [PATCH] make only one z
---
wganHCAL.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/wganHCAL.py b/wganHCAL.py
index a61ab5a..9835cc8 100644
--- a/wganHCAL.py
+++ b/wganHCAL.py
@@ -85,7 +85,8 @@ def train(args, aD, aG, aGE, device, train_loader, optimizer_d, optimizer_g, epo
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)
+ z = zE.view(args.batch_size, args.nz)
+ #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
@@ -136,7 +137,8 @@ def train(args, aD, aG, aGE, device, train_loader, optimizer_d, optimizer_g, epo
fake_ecal = fake_ecal.unsqueeze(1)
####
- z = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz))), requires_grad=True)
+ z = zE.view(args.batch_size, args.nz)
+ #z = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz))), requires_grad=True)
## generate fake data out of noise
fake_dataHCALG = aG(z, label, fake_ecal)
--
GitLab