From ac8e2c287c46df9e5de1e2b41eba630ae3bbcb4d Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Wed, 27 Jul 2022 12:04:55 +0200
Subject: [PATCH] dimension error fixed
---
wgan_ECAL_HCAL_3crit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/wgan_ECAL_HCAL_3crit.py b/wgan_ECAL_HCAL_3crit.py
index d3da95e..01b2f30 100644
--- a/wgan_ECAL_HCAL_3crit.py
+++ b/wgan_ECAL_HCAL_3crit.py
@@ -242,7 +242,7 @@ def train(args, aDE, aDH, aD_H_E, aGE, aGH, device, train_loader, optimizer_d_E,
fake_ecal = fake_ecal.unsqueeze(1).detach()
z = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz))), requires_grad=True)
- fake_dataHCAL = aGH(z, label.view(-1, 1, 1, 1, 1), fake_ecal) ## 48 x 30 x 30
+ fake_dataHCAL = aGH(z, label, fake_ecal) ## 48 x 30 x 30
fake_dataHCAL = fake_dataHCAL.unsqueeze(1)
## Loss function for ECAL generator
--
GitLab