1. Remove the gradient check in train
2. add eps in the cost function