Hi Nick,
Thanks for your contribution! The code is pretty neat!

Have you chosen a License? Would love to contribute to it.

If I may suggest some features I’d like to have re: model trainer:

GPU support
Ability to use PyTorch pretrained models (SuperModel can’t be used as a subclass for official VGG, Resnet, etc models)
Accuracy logging

I’ve already implemented 1 and 2 in my fork. I’d be happy to send a PR if you’re OK with those features.
Rodrigo