[GAN/conditional_gan]fix a mix use of tensor( size 64x1 ) and tensor(…

… size 64 )
wiseodd · Apr 3, 2018 · 866b046 · 866b046
1 parent c146d7d
commit 866b046
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GAN/conditional_gan/cgan_pytorch.py b/GAN/conditional_gan/cgan_pytorch.py
@@ -76,8 +76,8 @@ def reset_grad():
 G_solver = optim.Adam(G_params, lr=1e-3)
 D_solver = optim.Adam(D_params, lr=1e-3)
 
-ones_label = Variable(torch.ones(mb_size))
-zeros_label = Variable(torch.zeros(mb_size))
+ones_label = Variable(torch.ones(mb_size, 1))
+zeros_label = Variable(torch.zeros(mb_size, 1))
 
 
 for it in range(100000):