search

aitorzip / PyTorch-CycleGAN

A clean and readable Pytorch implementation of CycleGAN
https://arxiv.org/abs/1703.10593
GNU General Public License v3.0
1.22k stars 283 forks source link

Import Error! #3

Open happsky opened 6 years ago

happsky commented 6 years ago

Traceback (most recent call last): File "./train", line 6, in import torchvision.transforms as transforms ImportError: No module named 'torchvision'

abi98213 commented 6 years ago

I also faced the same issue, I deleted all the command line inputs and set the required values to default value so I just did "python ./train" and it solved this problem. But it is now giving error RuntimeError: input and target shapes do not match: input [1 x 1], target [1] at /pytorch/aten/src/THCUNN/generic/MSECriterion.cu:15 Don't know how to solve

sshkhr commented 6 years ago

@mabdullahrafique

Change the output of the discriminator to this : return F.avg_pool2d(x, x.size()[2:]).view(x.size()[0])