google-deepmind / google-deepmind/sonnet
how to remove model from gpu memory？
Open
- Dominant language
- Python
- Stars
- 10k
- Forks
- 1.3k
- PR merge metrics
- No merged PRs in 30d
Description
I have model_a and model_b, is there any elegant way to remove model_a from gpu memory after i copy weights from model_a to model_b. Except for save and reload from disk
Contributor guide
Assessment
This issue has not been assessed yet.