09 2021 档案

机器学习-完整的模型训练套路（pytorch环境）

摘要：一个例子 import torch import torchvision # 准备数据集 from torch import nn from torch.nn import Sequential, Conv2d, MaxPool2d, Flatten, Linear from torch.utils 阅读全文

posted @ 2021-09-05 16:55 EA2218764AB 阅读(229) 评论(0) 推荐(0) 编辑

机器学习-网络模型的保存与使用（pytorch环境）

摘要：例子 vgg16_false = torchvision.models.vgg16(pretrained=False) # 保存方式1（网络模型） torch.save(vgg16_false,"vgg_method1.pth") # 保存方式2（参数模型） torch.save(vgg16_fal 阅读全文

posted @ 2021-09-04 20:04 EA2218764AB 阅读(83) 评论(0) 推荐(0) 编辑