how to remove model from gpu memory?
Author: xfchen0912Created Mar 19, 2022Updated Mar 24, 2022
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
Source: google-deepmind/sonnet