Closed dp-isi closed 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
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