SGD seems to diverge