search

aharley / pips

Particle Video Revisited
MIT License
571 stars 51 forks source link

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

Closed ZHAOZHIHAO closed 1 year ago

ZHAOZHIHAO commented 1 year ago

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

aharley commented 1 year ago

Thanks! Fixed this in 486124b