transformers
8becb732 - removing torch.cuda.empty_cache() from TF function (#3267)

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
Author
Parents
Loading