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