From 300898409448d6e8bf2ee7719f64bf9388bcedc6 Mon Sep 17 00:00:00 2001
From: chenguowei01 <chenguowei01@baidu.com>
Date: Wed, 17 Jun 2020 11:03:59 +0800
Subject: [PATCH] update train.py

---
 dygraph/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dygraph/train.py b/dygraph/train.py
index abc55179..c8dce1b0 100644
--- a/dygraph/train.py
+++ b/dygraph/train.py
@@ -171,7 +171,7 @@ def train(model,
                 loss = model(images, labels, mode='train')
                 loss.backward()
             optimizer.minimize(loss)
-            model_parallel.clear_gradients()
+            model.clear_gradients()
             logging.info("[TRAIN] Epoch={}/{}, Step={}/{}, loss={}".format(
                 epoch + 1, num_epochs, step + 1, num_steps_each_epoch,
                 loss.numpy()))
-- 
GitLab