fix dataloader next

cddc899f · 悟、 · zengshao0622 · 4af0d9bb · cddc899f
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppcls/engine/train/train.py ppcls/engine/train/train.py +2 -2

未找到文件。
--- a/ppcls/engine/train/train.py
+++ b/ppcls/engine/train/train.py
@@ -28,10 +28,10 @@ def train_epoch(engine, epoch_id, print_batch_step):
    for iter_id in range(engine.iter_per_epoch):
        # fetch data batch from dataloader
        try:
-            batch = engine.train_dataloader_iter.next()
+            batch = next(engine.train_dataloader_iter)
        except Exception:
            engine.train_dataloader_iter = iter(engine.train_dataloader)
-            batch = engine.train_dataloader_iter.next()
+            batch = next(engine.train_dataloader_iter)

        profiler.add_profiler_step(engine.config["profiler_options"])
        if iter_id == 5: