Closed killuaZold closed 3 years ago
i fixed the problem
i think you should change the following code
transform = transforms.Compose([transforms.Resize((input_size, input_size)), transforms.ToTensor(), transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))])
transform = transforms.Compose([ transforms.Resize(32), transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,)) ])
i run the original code,but i get this problem.and i don't know why can you help me thx (👍