Skip to content
Snippets Groups Projects
Commit c0001547 authored by Engin Eren's avatar Engin Eren
Browse files

correcting the unused variable

parent 8be29b44
No related branches found
No related tags found
1 merge request!43crit peter
Pipeline #4181883 passed
Stage: build
Stage: test
Hide whitespace changes
Inline Side-by-side
Showing
with 5 additions and 2 deletions
wganHCAL.py
+ 5
− 2
View file @ c0001547
...@@ -68,10 +68,13 @@ def train(args, aD, aG, aGE, device, train_loader, optimizer_d, optimizer_g, epo ...@@ -68,10 +68,13 @@ def train(args, aD, aG, aGE, device, train_loader, optimizer_d, optimizer_g, epo
label = energy.to(device).float() 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 ## Generate Fake HCAL
z = Variable(Tensor(np.random.uniform(-1, 1, (args.batch_size, args.nz))), requires_grad=False) 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 fake_dataHCAL = aG(z, label, fake_ecal).detach() ## 48 x 30 x 30
## Critic fwd pass on Real ## Critic fwd pass on Real
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment