Add 'with torch.no_grad()' to integration test forward pass (#14821)

b37cf7de · Henrik Holm · GitHub · 952a77b0 · b37cf7de
Unverified Commit b37cf7de authored Dec 20, 2021 by Henrik Holm Committed by GitHub Dec 20, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/test_modeling_deberta.py tests/test_modeling_deberta.py +2 -1

No files found.
--- a/tests/test_modeling_deberta.py
+++ b/tests/test_modeling_deberta.py
@@ -275,6 +275,7 @@ class DebertaModelIntegrationTest(unittest.TestCase):

        input_ids = torch.tensor([[0, 31414, 232, 328, 740, 1140, 12695, 69, 46078, 1588, 2]])
        attention_mask = torch.tensor([[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]])
+        with torch.no_grad():
            output = model(input_ids, attention_mask=attention_mask)[0]
        # compare the actual values for a slice.
        expected_slice = torch.tensor(