tests/test_modeling_distilbert.py · 97f3beed3616d50c722fc9227205f8330048ca6c · chenpangpang / transformers

Add `with torch.no_grad()` to DistilBERT integration test forward pass (#14979) · 97f3beed

Jake Tae authored Jan 13, 2022

* refactor: wrap forward pass around no_grad context

* Update tests/test_modeling_distilbert.py

* fix: rm `no_grad` from non-integration tests

* chore: rm whitespace change

97f3beed

test_modeling_distilbert.py 11.9 KB

Replace test_modeling_distilbert.py