InstanceNorm2d throws an error

[golunovas]

Thank you for the great work.
I noticed that the training code of CFSM is failing with the input size 112 with the following error:
ValueError: Expected more than 1 spatial element when training, got input size torch.Size([32, 512, 1, 1])
Apparently, it's coming from the InstanceNorm2d https://github.com/liuf1990/CFSM/blob/main/CFSM/synthesis_network.py#L200

Any advice on how to solve it?

[arnav-roy]

The downsample is reducing the size. remove downsample(

CFSM/CFSM/synthesis_network.py

Line 229 in dd56184

x = self.downsample(x)

) or resize your input image to a higher size.