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

handle index error for dataloader

上级 ff81fa01
......@@ -12,3 +12,4 @@ build/
log/
nohup.out
.DS_Store
.idea
......@@ -371,6 +371,11 @@ def run(dataloader,
"Except RuntimeError when reading data from dataloader, try to read once again..."
)
continue
except IndexError:
logger.warning(
"Except IndexError when reading data from dataloader, try to read once again..."
)
continue
idx += 1
# ignore the warmup iters
if idx == 5:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册