I'm excited to introduce a custom PyTorch wrapper included in my Machine Learning Training Utilities. From now on, it will support TensorFlow and PyTorch implementations! I encourage you to check out my tutorial and try it out for yourself: pylessons.com/pytorch-wrapper