GitHub repository: vishalbelsare/torchbearer
Description: torchbearer: A model fitting library for PyTorch