Import Error!

[happsky]

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

[abi98213]

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]

@mabdullahrafique

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