摘要: pytorch中模型的保存pth,pkl,pt格式 pytorch保存模型非常简单,主要有两种方法: 只保存参数;(官方推荐) 保存整个模型 (结构+参数)。 由于保存整个模型将耗费大量的存储,故官方推荐只保存参数,然后在建好模型的基础上加载。本文介绍两种方法,但只就第一种方法进行举例详解。 模型保 阅读全文
posted @ 2021-09-08 18:04 编程coding小白 阅读(929) 评论(0) 推荐(0) 编辑