https://cognitivehub.in/pytorch-connection-between-loss-backward-and-optimizer-step/