multimodalart's picture
Update trainer.py
5b1ae02