custom-optimizers-pytorch Customizing optimizers like SGD, Adam in Pytorch for experiments to include loss term