Missmatch between Current network configure and pretrained weights

Hi, 

I am trying to run the inference, but I encounter an error when loading the state dictionary in GenGnn_standalone.py at line 71. How can I resolve this issue? Do I need to make any changes to the code?

Thanks,
