这种情况应该怎么办，train.py代码运行会报错，麻烦您解答一下，谢谢啦 
