PyCoach is a Python package that provides:
- A high-level abstraction for PyTorch model training;
- Well known callbacks structure based on Keras;
The main purpose of PyCoach is handle training, evaluation and prediction, leaving other tasks like create the network model and dataloaders to the users.
This is an initial personal project and contribution to it is appreciated.
Actual status: Working with PyTorch 0.4!
- Document the use of the package;
- Add use examples;
- Document the code;
- Complete Keras callbacks ports to PyCoach:
- CallbackList
- Callback
- BaseLogger
- TerminateOnNaN
- ProgbarLogger
- History
- ModelCheckpoint
- EarlyStopping
- RemoteMonitor
- LearningRateScheduler
- TensorBoard
- ReduceLROnPlateau
- CSVLogger
- LambdaCallback
- Plotter (new!)
- TextLogger (new!)
- Improve predict function;