search

ZhaoJ9014 / face.evoLVe

🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥
MIT License
3.44k stars 760 forks source link

fix iter bug #39

Closed zllrunning closed 5 years ago

zllrunning commented 5 years ago

It will always iter the first batch if we do like this.

        while idx + batch_size <= len(loader.dataset):
            batch, _ = iter(loader).next()

Maybe we should execute iter(loader) outside the while loop.