oh um small question what is the difference between Trainer.predict(your_test_dataset)  and Trainer.evaluate(your_test_dataset) ? I think that predict might solve my problem. i was using evaluate and getting errors previously.