add unit test for clear_param_diffs

andpol5 · Jun 22, 2016 · 892c78d · 892c78d
1 parent 118c97f
commit 892c78d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/python/caffe/test/test_net.py b/python/caffe/test/test_net.py
@@ -63,6 +63,17 @@ def test_forward_backward(self):
         self.net.forward()
         self.net.backward()
 
+    def test_clear_param_diffs(self):
+        # Run a forward/backward step to have non-zero diffs
+        self.net.forward()
+        self.net.backward()
+        diff = self.net.params["conv"][0].diff
+        # Check that we have non-zero diffs
+        self.assertTrue(diff.max() > 0)
+        self.net.clear_param_diffs()
+        # Check that the diffs are now 0
+        self.assertTrue((diff == 0).all())
+
     def test_inputs_outputs(self):
         self.assertEqual(self.net.inputs, [])
         self.assertEqual(self.net.outputs, ['loss'])