文章分类 -  Torch

摘要:转载:https://www.zhihu.com/question/319682377 一 pytorch模型保存与加载 模型包括模型结构和模型参数,因此,有不同的保存模型的方式。 1 官方推荐 1.1 只保存模型参数 优点:只保存模型参数,该方法速度快,占用的空间少 缺点:重载模型时,必须知道模型 阅读全文
posted @ 2022-06-17 18:22 IllidanStormrage 阅读(536) 评论(0) 推荐(1) 编辑