Update DNN_train.cpp