I'm trying to change the input size from 256x256 to 32x32 in generator model but comes up with an error

[monsieurupanshu]

Error
raise ValueError("Expected more than 1 value per channel when training, got input size {}".format(size)) ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

I think I'm doing something wrong.