SheldonGeek

You can change transforms.Normalize to transforms.Normalize((0.5), (0.5)) in cp_dataset.py. It will fix it. I had the same issue. and it got fixed. The error happened because the first dimension of...