transformers
5aec50ec - When save a model on TPU, make a copy to be moved to CPU (#27993)

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