Typo in train.py, train_iterloader = iter(val_dataloader)
ZHAOZHIHAO opened this issue · comments
Hi,
Line 374 in train.py seems should be "val_iterloader = iter(val_dataloader)" instead of "train_iterloader = iter(val_dataloader)".
gotit = (False,False)
while not all(gotit):
try:
sample, gotit = next(val_iterloader)
except StopIteration:
train_iterloader = iter(val_dataloader) # line 374
sample, gotit = next(val_iterloader)
Best
Thanks! Fixed this in 486124b