GYQ-haha

Results 2 issues of GYQ-haha

在运行 train.py 时出现了这个问题,提示错误在train.py的第120行batch_iterator = iter(train_dataloader)这句话,请问如何解决呢?

![image](https://user-images.githubusercontent.com/56821870/142424086-0c95c0d4-d15b-4304-80a1-0b8c7b465566.png)