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

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

Browse Source
This commit is contained in:
Partho
2022-10-04 19:43:22 +05:30
committed by GitHub
parent d6e920449e
commit a9782881a4
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/models/big_bird/test_modeling_big_bird.py
View File

@@ -627,7 +627,8 @@ class BigBirdModelIntegrationTest(unittest.TestCase):
model.to(torch_device) model.to(torch_device)
input_ids = torch.tensor([[20920, 232, 328, 1437] * 1024], dtype=torch.long, device=torch_device) input_ids = torch.tensor([[20920, 232, 328, 1437] * 1024], dtype=torch.long, device=torch_device)
outputs = model(input_ids) with torch.no_grad():
outputs = model(input_ids)
prediction_logits = outputs.prediction_logits prediction_logits = outputs.prediction_logits
seq_relationship_logits = outputs.seq_relationship_logits seq_relationship_logits = outputs.seq_relationship_logits
@@ -655,7 +656,8 @@ class BigBirdModelIntegrationTest(unittest.TestCase):
model.to(torch_device) model.to(torch_device)
input_ids = torch.tensor([[20920, 232, 328, 1437] * 512], dtype=torch.long, device=torch_device) input_ids = torch.tensor([[20920, 232, 328, 1437] * 512], dtype=torch.long, device=torch_device)
outputs = model(input_ids) with torch.no_grad():
outputs = model(input_ids)
prediction_logits = outputs.prediction_logits prediction_logits = outputs.prediction_logits
seq_relationship_logits = outputs.seq_relationship_logits seq_relationship_logits = outputs.seq_relationship_logits
@@ -920,7 +922,8 @@ class BigBirdModelIntegrationTest(unittest.TestCase):
model.eval() model.eval()
input_ids = torch.tensor([200 * [10] + 40 * [2] + [1]], device=torch_device, dtype=torch.long) input_ids = torch.tensor([200 * [10] + 40 * [2] + [1]], device=torch_device, dtype=torch.long)
output = model(input_ids).to_tuple()[0] with torch.no_grad():
output = model(input_ids).to_tuple()[0]
# fmt: off # fmt: off
target = torch.tensor( target = torch.tensor(