![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
lai199508
这个作者很懒,什么都没留下…
展开
-
Resnet for Fashion_Mnist(一)
Resnet for Fashion_Mnist目录写作目的Fashion_Mnist制作数据集写Datasets函数模型定义模型训练模型加载及预测数据 目录 写作目的 通过使用resnet 实现 Fashion_Mnist,学会使用pytorch框架。这篇博客全方位介绍了如何使用pytorch,包括数据制作、模型定义、模型训练及验证,模型保存、模型加载、测试集预测。 Fashion_Mnist ...原创 2018-12-08 17:13:56 · 1475 阅读 · 0 评论 -
Resnet for Fashion_Mnist(四)
Resnet for Fashion_Mnist(五) 模型加载及预测数据 下面给出模型加载及预测数据的代码,预测结果保存到csv文件中。 # -*- coding: utf-8 -*- import torch import MyDataSet from torchvision import transforms from torch.autograd import Variable impor...原创 2018-12-16 19:46:36 · 438 阅读 · 0 评论 -
Resnet for Fashion_Mnist(三)
Resnet for Fashion_Mnist(四) 模型训练 下面给出模型训练、验证的代码,完成后保存模型。 import torch from torchvision import datasets, models, transforms import torch.optim as optim import model import utils import time import argp...原创 2018-12-16 19:48:32 · 507 阅读 · 0 评论 -
Resnet for Fashion_Mnist(二)
Resnet for Fashion_Mnist(三) 模型定义 下面给出Resnet的模型定义代码,包含了resnet18。想写其他restnet模型,可以在此基础上扩展。 import torch import torch.nn as nn import utils import math from torch.autograd import Variable __all__ = ['Re...原创 2018-12-16 19:49:56 · 901 阅读 · 0 评论