Mismatch error on Generator.py

[anthorax1234]

There seems to be an issue with the generator.py code after training. When trying to load the state dict, it gives a mismatch error on the size of the param/model.

[Rik Koncel-Kedziorski]

I made changes to the repo today, does this error still happen?

[anthorax1234]

To test properly, what should the "-data" flag be set to?

[Rik Koncel-Kedziorski]

set -data to your test data (i.e. preprocessed.test.tsv) and if you are using a custom dataset, then set the -traindata flag to your training data file

[anthorax1234]

Everything is working perfectly now. Thanks a lot for the quick replies!