fix bug. iter start from 0 when resume

438e7310 · chenguowei01 · f4ee7706 · 438e7310
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

dygraph/core/train.py dygraph/core/train.py +1 -1

未找到文件。
--- a/dygraph/core/train.py
+++ b/dygraph/core/train.py
@@ -79,7 +79,7 @@ def train(model,
    train_batch_cost = 0.0
    timer.start()
-    iter = 0
+    iter = start_iter
    while iter < iters:
        for data in loader:
            iter += 1