From 866b04614eda4c5d27aedd60ddab9437b2edcd86 Mon Sep 17 00:00:00 2001 From: mayorx Date: Tue, 3 Apr 2018 17:46:54 +0800 Subject: [PATCH] [GAN/conditional_gan]fix a mix use of tensor( size 64x1 ) and tensor( size 64 ) --- GAN/conditional_gan/cgan_pytorch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/GAN/conditional_gan/cgan_pytorch.py b/GAN/conditional_gan/cgan_pytorch.py index d83f2d6..703762c 100644 --- 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):