transformers
f634ba14
- fix tensor to gpu
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
fix tensor to gpu
References
add_inference_caching_reformer
#5578 - [Reformer] - Cache hidden states and buckets to speed up inference
Author
patrickvonplaten
Parents
07d7ff65
Loading