2019年11月26日
深度之眼PyTorch训练营第二期---16、模型保存与加载
摘要： 一、序列化与反序列化 1、torch.save 主要参数： obj：对象 f：输出路径 2、torch.load 主要参数： f：文件路径 map_location：指定存放位置，cpu or gpu 二、模型保存与加载的两种方式 第一种方式： 保存整个Module torch.save(net,p 阅读全文
posted @ 2019-11-26 22:24 cola_cola 阅读(228) 评论(0) 推荐(0) 编辑