Update FirstTrain_MLP_Linear.py