fix sgd update equation