fix backward in tests

This commit is contained in:
thomwolf
2019-12-10 17:50:32 +01:00
parent f2538c1274
commit 67a8be8e90

View File

@@ -277,8 +277,7 @@ class CommonTestCases:
inputs = inputs_dict.copy()
inputs['head_mask'] = head_mask
with torch.no_grad():
outputs = model(**inputs)
outputs = model(**inputs)
# Test that we can get a gradient back for importance score computation
output = sum(t.sum() for t in outputs[0])