Error with Inference

[akmaralAW]

Hello !

I am getting RuntimeError: Error(s) in loading state_dict for DataParallel: size mismatch for module.main_decoder.upsample error. What may cause this error ? I am using images of the same size as for training dataset images.

Akmaral

[yassouali]

Hi @akmaralAW

Maybe try commenting line 85 (not adding data-parallel)

[akmaralAW]

Great ! Works ! Thanks !