Coding SGD with Momentum Optimizer in PyTorch: Step-by-Step Guide - MRELAN