Fixed using the wrong parameter for the starting pointof the SGD algorithm