有个小建议，能否更新一下train代码

import logging
logger = logging.getLogger() # logging对象
#fh = logging.FileHandler("test.log") # 文件对象
sh = logging.StreamHandler() # 输出流对象
fm = logging.Formatter('%(asctime)s-%(message)s') # 格式化对象
#fh.setFormatter(fm) # 设置格式
sh.setFormatter(fm) # 设置格式
#logger.addHandler(fh) # logger添加文件输出流
logger.addHandler(sh) # logger添加标准输出流（std out）
logger.setLevel(logging.DEBUG) # 设置从那个等级开始提示

因为我用python3 没有INFO 信息，找了半天没找到原因，发现是 py3的logging 代码需要改进