Open ZhengWwwq opened 3 days ago
2.在utils.py的DataLoader类中的next方法中,在train模式下,似乎没有raise StopIteration(那行注释是我自己加的),这样就会在for batch in self.train_loader时,无限循环(可以通过加一层判断来终止)
已修复,请同步仓库
这里在train模式下Dataloader会无限循环是因为我们使用训练步数来控制迭代,而不是epoch-batch模式,请检查TrainerR.train()的while循环条件,不必使用下面的语句:
train
Dataloader
TrainerR.train()
while
for batch in self.train_loader:
请直接使用Dataloader的next方法
next
2.在utils.py的DataLoader类中的next方法中,在train模式下,似乎没有raise StopIteration(那行注释是我自己加的),这样就会在for batch in self.train_loader时,无限循环(可以通过加一层判断来终止)