The main intention of this library is to be a light and flexible script to easily train pytorch models. It has the options for checkpointing, logging with tensorboard, visualisation and more.
To install simply download this repo. Go to the root directory i.e Trainer/ and enter the following commmand
pip install -e .
Notice the "." at the end of command!