Update sgd.py

no need for rng