Pytorch training wrapper library? (like Tensorpack?)

1

Tensorflow has Tensorpack which really streamlines data loading, checkpointing, model training, saving, multi-gpu training.


Author for library above works in Facebook Research. Does @smth know him?

Anyone started something similar with Pytorch?

1 Like
2

Maybe Ignite or PyTorchNet is what you are looking for.

3

Iā€™m using this package for different features (2d affine transformations),
however - it appears to have high level training wrappers, allows callbacks, and few other things (seems highly influenced by keras)

perhaps it will suit your needs.