Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add
with torch.no_grad()
to DistilBERT integration test forward pass (
huggingface#14979) * 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
- Loading branch information