wrap forward passes with torch.no_grad() (#19278)

2022-10-04 19:39:23 +05:30
parent f134d38553
commit 2403dbd607
1 changed files with 2 additions and 1 deletions
--- a/tests/models/deberta_v2/test_modeling_deberta_v2.py
+++ b/tests/models/deberta_v2/test_modeling_deberta_v2.py
@@ -299,6 +299,7 @@ class DebertaV2ModelIntegrationTest(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(