torchtricks

PyTorch trainers, datasets, criterion, and utility functions


License
MIT
Install
pip install torchtricks==0.0.1

Documentation

kerasine

[WIP] A collection of PyTorch trainers, datasets, criterion, and utility functions

  • Add TQDM and TQDM notebook Callbacks
  • Add reporting with TensorBoard Callback
  • Add Metric callbacks
  • Support schedulers
  • Allow easy subclassing for complex fit routines.