try pytorch-lightning

[mxochicale]

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
https://github.com/PyTorchLightning/pytorch-lightning

[mxochicale]

i close this one as for now i need to understand the fundamentals of pytorch