SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Wrap RemBert integration test forward passes with torch.no_grad() (#21503)

Browse Source 
added with torch.no_grad() to the integration tests and applied make style

Co-authored-by: Bibi <Bibi@katies-mac.local>
This commit is contained in:
Katie Le
2023-02-08 08:00:52 -05:00
committed by GitHub
parent 5b67ab9924
commit cc1d0685b3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/rembert/test_modeling_rembert.py
View File

@@ -464,7 +464,8 @@ class RemBertModelIntegrationTest(unittest.TestCase):
model = RemBertModel.from_pretrained("google/rembert")
input_ids = torch.tensor([[312, 56498, 313, 2125, 313]])
segment_ids = torch.tensor([[0, 0, 0, 1, 1]])
output = model(input_ids, token_type_ids=segment_ids, output_hidden_states=True)
with torch.no_grad():
output = model(input_ids, token_type_ids=segment_ids, output_hidden_states=True)
hidden_size = 1152