https://pankaj-tiwari2.medium.com/stochastic-gradient-descent-using-pytorch-linear-module-c771a22f6493