PyTorch
加油_加油_DrDu
科研是一种乐趣
展开
-
【PyTorch】卷积神经网络LeNet5-训练Fashion-MNIST
使用fashion-MNIST演示PyTorch实现卷积神经网络LeNet5的创建、训练和测试导入依赖包import torchimport torch.nn as nnimport torchvisionimport torch.utils.data as Dataimport torchvision.transforms as transformsimport sys加载数据集mnist_train = torchvision.datasets.FashionMNIST(root='原创 2020-08-12 17:24:55 · 1737 阅读 · 0 评论 -
【PyTorch】多层感知机-训练Fashion-MNIST
使用fashion-MNIST演示PyTorch实现多层感知的创建、训练和测试导入依赖包import torchimport torch.utils.data as Dataimport torchvisionimport torchvision.transforms as transformsimport torch.nn as nnfrom torch.nn import initimport sys加载数据集mnist_train = torchvision.datasets.F原创 2020-08-12 16:55:25 · 650 阅读 · 0 评论 -
【PyTorch】模型构造:Module 和其子类Sequential、ModuleList、ModuleDict
0. 导入用到的库import torchimport torch.nn as nnfrom collections import OrderedDict1. 继承Module构造和访问模型Module 类是 torch.nn 模块里提供的⼀个模型构造类,是所有神经⽹络模块的基类,我们可以继承它来定义我们想要的模型。只需重写 forward方法class MLP(nn.Module): def __init__(self, **kwargs): su.原创 2020-08-11 15:07:04 · 520 阅读 · 0 评论 -
PyTorch模型保存和加载以便继续训练
通用的PyTorch模型保存和加载模板保存模型state = { 'model': model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch': epoch } torch.save(state, path)加载模型checkpoint = torch.load(path)model.load_state_dict(checkpoint['model'])optimizer.load_state_dict(checkpoin原创 2020-08-11 13:09:32 · 967 阅读 · 0 评论