RuntimeError: input and target shapes do not match: input [1 x 1 x 1 x 1], target [1 x 1] about mgan HOT 5 CLOSED

Hi@ @hondahmd , I met the same problem, could you please tell me how to solve this

from mgan.

@buaaswf Sorry I didn't figure out a solution. But I believe that it might have something to do with your develop environment, your package version, for example.

from mgan.

@buaaswf Sorry I didn't figure out a solution. But I believe that it might have something to do with your develop environment, your package version, for example.

I also have the same problem. As you said, it has something to do with the development environment version, as show below:
input and target shapes do not match: input [1 x 1 x 1 x 1], target [1 x 1] at c:\programdata\miniconda3\condabld\pytorch_1533086611947\work\aten\src\thcunn\generic/MSECriterion.cu:12
So, have you solved it yet?

from mgan.

modify the line 71-74 in train.py :
real_lbl=torch.zeros(batch_size,1,1,1).cuda() real_lbl[:,0,0,0]=1 fake_lbl=torch.zeros(batch_size,1,1,1).cuda() fake_lbl[:,0,0,0]=1 @hondahmd

from mgan.

modify the line 71-74 in train.py :
real_lbl=torch.zeros(batch_size,1,1,1).cuda() real_lbl[:,0,0,0]=1 fake_lbl=torch.zeros(batch_size,1,1,1).cuda() fake_lbl[:,0,0,0]=1 @hondahmd

Thx! It works!

from mgan.