v2.2-modify the para of train.py