transformers
97f3beed - Add `with torch.no_grad()` to DistilBERT integration test forward pass (#14979)
Commit
View On GitHub
Commit
4 years ago
Add `with torch.no_grad()` to DistilBERT integration test forward pass (#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
Author
jaketae jaketae
Parents
021f2ea9
FAQ Terms Privacy Refunds Impressum
Loading