Mismatch error on Generator.py
anthorax1234 opened this issue · comments
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.
I made changes to the repo today, does this error still happen?
To test properly, what should the "-data" flag be set to?
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
Everything is working perfectly now. Thanks a lot for the quick replies!