handle index error for dataloader
parent
ff81fa0179
commit
0d2769d2d4
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue