add print to debug

PaddlePaddle · May 4, 2022 · 3a21267 · 3a21267
1 parent f353873
commit 3a21267
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_optimizer.py b/python/paddle/fluid/tests/unittests/test_optimizer.py
@@ -87,14 +87,18 @@ def forward(self, x):
             asgd = paddle.optimizer.ASGD(learning_rate=1., parameters=model.parameters(), t0=1)
 
             loss = model(x)
+            print(f'1: w grad before bw: {model._w.grad}')
             loss.backward()
+            print(f'1: w grad: {model._w.grad}')
             asgd.step()
             assert np.allclose(model._w.numpy(), np_neg_ones)
             assert np.allclose(asgd.averaged_parameters()[0].numpy(), np_neg_ones)
             asgd.clear_grad()
 
             loss = model(x)
+            print(f'2: w grad before bw: {model._w.grad}')
             loss.backward()
+            print(f'2: w grad: {model._w.grad}')
             asgd.step()
             assert np.allclose(model._w.numpy(), np_neg_ones * 2)
             assert np.allclose(asgd.averaged_parameters()[0].numpy(), np_neg_ones * 2)