Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "6a08162ad4ef2e08de45301cc9197f2bee01cbef"
Unverified Commit df2f2812 authored by Partho's avatar Partho Committed by GitHub
Browse files

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

parent 5f5e264a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/models/flaubert/test_modeling_flaubert.py
View file @ df2f2812
...@@ -437,7 +437,8 @@ class FlaubertModelIntegrationTest(unittest.TestCase): ...@@ -437,7 +437,8 @@ class FlaubertModelIntegrationTest(unittest.TestCase):
def test_inference_no_head_absolute_embedding(self): def test_inference_no_head_absolute_embedding(self):
model = FlaubertModel.from_pretrained("flaubert/flaubert_base_cased") model = FlaubertModel.from_pretrained("flaubert/flaubert_base_cased")
input_ids = torch.tensor([[0, 345, 232, 328, 740, 140, 1695, 69, 6078, 1588, 2]]) input_ids = torch.tensor([[0, 345, 232, 328, 740, 140, 1695, 69, 6078, 1588, 2]])
output = model(input_ids)[0] with torch.no_grad():
output = model(input_ids)[0]
expected_shape = torch.Size((1, 11, 768)) expected_shape = torch.Size((1, 11, 768))
self.assertEqual(output.shape, expected_shape) self.assertEqual(output.shape, expected_shape)
expected_slice = torch.tensor( expected_slice = torch.tensor(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment