如何从 GPU 内存中删除模型？

作者: xfchen0912创建于 2022年3月19日更新于 2022年3月24日

我有 model_a 和 model_b,有没有一种优雅的方法来在将权重从 model_a 复制到 model_b 之后从 GPU 内存中删除 model_a?除了从磁盘保存和重新加载之外

内容来源: google-deepmind/sonnet