add comment where I made a mistake
sgd version train