lai199508-CSDN博客

原创 Resnet for Fashion_Mnist（一）

Resnet for Fashion_Mnist目录写作目的Fashion_Mnist制作数据集写Datasets函数模型定义模型训练模型加载及预测数据目录写作目的通过使用resnet 实现 Fashion_Mnist，学会使用pytorch框架。这篇博客全方位介绍了如何使用pytorch，包括数据制作、模型定义、模型训练及验证，模型保存、模型加载、测试集预测。Fashion_Mnist...

2018-12-08 17:13:56 1470

原创 Resnet for Fashion_Mnist（二）

Resnet for Fashion_Mnist（三）模型定义下面给出Resnet的模型定义代码，包含了resnet18。想写其他restnet模型，可以在此基础上扩展。import torchimport torch.nn as nnimport utilsimport mathfrom torch.autograd import Variable__all__ = ['Re...

2018-12-16 19:49:56 891

原创 Resnet for Fashion_Mnist（三）

Resnet for Fashion_Mnist（四）模型训练下面给出模型训练、验证的代码，完成后保存模型。import torchfrom torchvision import datasets, models, transformsimport torch.optim as optimimport modelimport utilsimport timeimport argp...

2018-12-16 19:48:32 500

原创 Resnet for Fashion_Mnist（四）

Resnet for Fashion_Mnist（五）模型加载及预测数据下面给出模型加载及预测数据的代码，预测结果保存到csv文件中。# -*- coding: utf-8 -*-import torchimport MyDataSetfrom torchvision import transformsfrom torch.autograd import Variableimpor...

2018-12-16 19:46:36 429

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人