Typo in train.py, train_iterloader = iter(val_dataloader)

[zhihao zhao]

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

[Adam W. Harley]

Thanks! Fixed this in 486124b