• Sylvain Gugger's avatar
    Add tests to Trainer (#6605) · 573bdb0a
    Sylvain Gugger authored
    * Add tests to Trainer
    
    * Test if removing long breaks everything
    
    * Remove ugly hack
    
    * Fix distributed test
    
    * Use float for number of epochs
    573bdb0a
test_trainer.py 8.86 KB