From 1789da6422cde2af18c40f6c3de037fd744f68bb Mon Sep 17 00:00:00 2001 From: weishengyu Date: Mon, 18 Apr 2022 11:26:32 +0800 Subject: [PATCH] fix bug --- ppcls/engine/train/train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/ppcls/engine/train/train.py b/ppcls/engine/train/train.py index 3b02bac8..b15c1088 100644 --- a/ppcls/engine/train/train.py +++ b/ppcls/engine/train/train.py @@ -21,7 +21,6 @@ from ppcls.utils import profiler def train_epoch(engine, epoch_id, print_batch_step): tic = time.time() - v_current = [int(i) for i in paddle.__version__.split(".")] for iter_id, batch in enumerate(engine.train_dataloader): if iter_id >= engine.max_iter: break -- GitLab