simple-framework

Simple training framework for pytorch, with basic training loops and horovod support


License
MIT
Install
pip install simple-framework==0.0.1.1

Documentation

Training_framework

Training framework for pytorch models

Simple training framework for pytorch supporting two cases:

  • Normal training
  • Horovod Training

Use example: Open In Colab