2021年6月23日
pytorch-模型保存和加载
摘要： pytorch-模型保存和加载 加载模型参数和选择是由保存的模型数据结构决定，故先要确定保存模型模型的方法和数据结构 保存模型 # 模型权重参数 model.state_dict() '''首先说一下 model.state_dict() pytorch 中的 model.state_dict 是一 阅读全文
posted @ 2021-06-23 21:20 贝壳里的星海 阅读(1096) 评论(0) 推荐(0) 编辑