issue in train.py and test.py

sergeywong / cp-vton

Reimplemented code for "Toward Characteristic-Preserving Image-based Virtual Try-On Network"

MIT License

474 stars 182 forks source link

issue in train.py and test.py #13

Closed dp-isi closed 5 years ago

dp-isi commented 5 years ago

in train.py RuntimeError: The expanded size of the tensor (1) must match the existing size (3) at non-singleton dimension 0

in test.py: RuntimeError: The expanded size of the tensor (1) must match the existing size (3) at non-singleton dimension 0

dp-isi commented 5 years ago

in train.py RuntimeError: The expanded size of the tensor (1) must match the existing size (3) at non-singleton dimension 0

in test.py: RuntimeError: The expanded size of the tensor (1) must match the existing size (3) at non-singleton dimension 0 changing to torchvision 0.2.1 solved my issue sudo pip3 install torchvision==0.2.1