2021年11月24日
摘要: pytorch保存模型非常简单,主要有两种方法: 只保存参数;(官方推荐) 保存整个模型 (结构+参数)。 由于保存整个模型将耗费大量的存储,故官方推荐只保存参数,然后在建好模型的基础上加载。本文介绍两种方法,但只就第一种方法进行举例详解。 1、只保存参数 1)保存 一般地,采用一条语句即可保存参数 阅读全文
posted @ 2021-11-24 19:59 满船清梦压星河1024 阅读(659) 评论(0) 推荐(0)