Open takeru1205 opened 5 years ago
if you run train and get this error
input and target shapes do not match: input [1 x 1], target [1] at /pytorch/
you may change your code in train from
loss_GAN_B2A = criterion_GAN(pred_fake, target_real)
to
loss_GAN_B2A = criterion_GAN(pred_fake.squeeze(1), target_real)
other similar codes are also changed so.
Now if I run this utils.py and get this error
So utils.py update in line 39 and line 41
to