multimodalart's picture
Update trainer.py
b79f280