https://colab.research.google.com/drive/19YwLLREPmE0ZFoneEKQpSk-XRIIF9pPP#scrollTo=NxSD3yPDfEIj
https://www.deeplearningwizard.com/deep_learning/practical_pytorch/pytorch_feedforward_neuralnetwork/#step-4-instantiate-model-class