提交 0d2769d2 编写于 作者: L liuyuang

handle index error for dataloader

上级 ff81fa01
...@@ -12,3 +12,4 @@ build/ ...@@ -12,3 +12,4 @@ build/
log/ log/
nohup.out nohup.out
.DS_Store .DS_Store
.idea
...@@ -371,6 +371,11 @@ def run(dataloader, ...@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册