transformers
8becb732
- removing torch.cuda.empty_cache() from TF function (#3267)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
removing torch.cuda.empty_cache() from TF function (#3267) torch.cuda.empty_cache() was being called from a TF function (even when torch is unavailable) not sure any replacement is needed if TF OOMs
References
#3267 - removing torch.cuda.empty_cache() from TF function
Author
keskarnitish
Parents
ecfd3363
Loading