Mini-Batch SGD with PyTorch