transformers
5aec50ec
- When save a model on TPU, make a copy to be moved to CPU (#27993)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
When save a model on TPU, make a copy to be moved to CPU (#27993) * When save a model, make a copy to be moved to CPU, dont move the original model * make deepcopy inside of _save_tpu * Move to tpu without copy
References
#27993 - When save a model on TPU, make a copy to be moved to CPU
Author
qihqi
Parents
4edffda6
Loading