Welcome to Yingxin's GitHub page.

Hi 是完整的可以用

@semchan Hi 您好，目前没有添加batch的计划，以后有时间会试一下的。我最多只跑过3个epoch，精度比官方数据低，如果跑的时间长一点可能会得到更好的结果

Hello，感谢您的反馈！可以试一下最新的train.py，主要改了epoch_loss += Loss.item()，加过您qq啦