Open Mi5ke opened 4 years ago
Hi! This happens in latest versions of pytorch. You need to change transform function in dataset class from
self.transform = transforms.Compose([ \ transforms.ToTensor(), \ transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
to
self.transform = transforms.Compose([ \ transforms.ToTensor(), \ transforms.Normalize((0.5,), (0.5,))])
Hi,
I have downloaded all the data with
data_download.py
, but...When I try to train CP-VTON GMM stage, I get this error immediately:
RuntimeError: output with shape [1, 256, 192] doesn't match the broadcast shape [3, 256, 192]
See the full log below. Any help much appreciated! (I get the same error if I run cp_dataset.py alone.)
Thanks,
Mike