handle index error for dataloader

zengshao0622 · Jun 7, 2022 · 0d2769d · 0d2769d
1 parent ff81fa0
commit 0d2769d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -12,3 +12,4 @@ build/
 log/
 nohup.out
 .DS_Store
+.idea
diff --git a/ppcls/static/program.py b/ppcls/static/program.py
@@ -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:
-Original file line number
+Diff line change
@@ Expand Up / @@ -12,3 +12,4 @@ build/ @@
     log/
     nohup.out
     .DS_Store
+    .idea