Fix grad params

pytorch · Apr 5, 2021 · 1b19bcb · 1b19bcb
1 parent 23ebbc8
commit 1b19bcb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/attr/layer/test_layer_gradient_x_activation.py b/tests/attr/layer/test_layer_gradient_x_activation.py
@@ -113,8 +113,9 @@ def test_gradient_activation_embedding_no_grad(self) -> None:
         input1 = torch.tensor([2, 5, 0, 1])
         input2 = torch.tensor([3, 0, 0, 2])
         model = BasicEmbeddingModel()
-        # for param in model.parameters():
-        #    param.requires_grad = False
+        for param in model.parameters():
+            param.requires_grad = False
+
         with torch.no_grad():
             layer_act = LayerGradientXActivation(model, model.embedding1)
             self.assertEqual(