Fix train.py
parameter groups desc error
#6318
Merged
train.py
parameter groups desc error
#6318