IndexError: When performing advanced indexing the indexing objects must be LongTensors or convertible to LongTensors

[ptrblck]

This post and thread might hepp get rid of the module. names in your state_dict.