when i run the train.py, there are sth wrong with the codes, 
when i run the train.py, there are sth wrong with the codes,
