2022年8月11日
pytorch学习笔记（10）--完整的模型训练（待完善）
摘要： 一、神经网络训练 # file : train.py # time : 2022/8/11 上午10:03 # function : import torchvision.datasets from model import * from torch.utils.data import DataLo 阅读全文
posted @ 2022-08-11 11:11 helloWorldhelloWorld 阅读(71) 评论(0) 推荐(0)
pytorch学习笔记（9）--神经网络模型的保存与读取
摘要： 一、网络模型的保存和加载 1、网络模型保存方法1 import torch import torchvision vgg16 = torchvision.models.vgg16(weights=False) # 保存方法1：模型结构+模型参数 torch.save(vgg16, "vgg16_me 阅读全文
posted @ 2022-08-11 09:16 helloWorldhelloWorld 阅读(669) 评论(0) 推荐(0)