Generate RGB Images

Hello I would like to ask if SPADE takes fixed dimension maps because I am trying to use your pre trained model for COCO Stuff dataset to generate RGB Images for my own Sementic maps but i got that error We got that error complaining about the dimensions which really different form the input that SPADE use RuntimeError: Given groups=1, weight of size [1024, 184, 3, 3], expected input[1, 186, 8, 8] to have 184 channels, but got 186 channels instead.

The dimension of the semantic maps that i am using : 256*256. I would like to know the properties of the semantic maps that SPADE used to generate RGB Images. and if you can recommend me any way to solve this problem.

Thanks