2021年12月7日
pytorch加载网络模型（另外一个py文件加载网络模型）
摘要： 方式一：直接保存.pkl / .pth torch.save(model,'model.pth') torch.load('model.pth') 但是不推荐使用，因为本人实践还需要再代码开头添加import ... 导入py文件中的类模型 方法二：保存模型参数，推荐使用。 在另外一个py文件中构建 阅读全文
posted @ 2021-12-07 22:59 爱吃牛牛的康康 阅读(850) 评论(0) 推荐(0)