PyTorch Components for training neural networks.
Training Loop
Model Save, load

Code - https://colab.research.google.com/drive/1N7rH5cvUDnS_Tt_042y-fKE9p7bNASa2?usp=sharing