Minor tweak
Added a few more assert statements to dyn SGD