mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

handle index error for dataloader

pull/1988/head
liuyuang 2022-06-07 11:26:15 +08:00
parent ff81fa0179
commit 0d2769d2d4
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -12,3 +12,4 @@ build/
log/ log/
nohup.out nohup.out
.DS_Store .DS_Store
.idea

5
ppcls/static/program.py
View File

@ -371,6 +371,11 @@ def run(dataloader,
"Except RuntimeError when reading data from dataloader, try to read once again..." "Except RuntimeError when reading data from dataloader, try to read once again..."
) )
continue continue
except IndexError:
logger.warning(
"Except IndexError when reading data from dataloader, try to read once again..."
)
continue
idx += 1 idx += 1
# ignore the warmup iters # ignore the warmup iters
if idx == 5: if idx == 5: