Merge pull request #54 from mayorx/fix_warning

Cool! Thanks!
wiseodd · Apr 5, 2018 · c790d2c · c790d2c
2 parents 751cc1b + 866b046
commit c790d2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 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):

diff --git a/GAN/vanilla_gan/gan_pytorch.py b/GAN/vanilla_gan/gan_pytorch.py
@@ -74,8 +74,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):