正文

[基于pytorch的mnist识别03]运行模型(代码片段)

AIplusX  AIplusX  2023-01-11  332

关键词：

写在前面

今天调通了pytorch模型，同时进行了简单的模型测试

知识点总结

1：python忽略某些特定语句的warning：

import warnings
        with warnings.catch_warnings():#ignore some warnings
            warnings.simplefilter("ignore")
            loss = criterion(outputs, labels)  # calculate loss

2：pytorch将cpu上的数据移动到gpu上：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
device
mnist_net.to(device)
print(next(mnist_net.parameters()).device)

        images = images.to(device)
        labels = labels.to(device)

3：函数形参位置，带有默认值的参数应该放在没有默认值参数的后面，否则会报错的；

4：torch.max()用法：

index, parameter = torch.max(para，0)，返回列最大值以及对应索引，索引在前，数值在后

index, parameter = troch.max(data，1)，返回行最大值以及对应索引，索引在前，数值在后

5：pytorch在单行内进行刷新输出：

import sys

for i, (imgs, labs) in enumerate(usermnist_validate_loader):
	sys.stdout.write('\\r'+str(i)+str())

6：pytorch报错：

“RuntimeError: expected scalar type Float but found Double”
这个错误是数据类型不统一引起的。

解决方法：

torch.set_default_tensor_type(torch.DoubleTensor)

这个语句主要是设置tensor的默认数据类型

主要内容

主要内容在我的古月居博客：
[基于Pytorch的MNIST识别03]运行模型

明天就可以真正开始调参了。

[基于pytorch的mnist识别05]总结(代码片段)

写在前面走走停停，我自己做的第一个机器学习的项目：MNIST手写字符集的识别，终于结束了，一路走来也算是踩了大大小小的坑，在这篇文章里做一个总结。模型配置模型总共有4层，输入层有784个神经元... 查看详情

pytorch基于cnn的手写数字识别（在mnist数据集上训练）(代码片段)

最终成果http://pytorch-cnn-mnist.herokuapp.com/GITHUBhttps://github.com/XavierJiezou/pytorch-cnn-mnist本文以最经典的mnist数据集为例，讲述了使用pytorch做机器学习的一整套流程，文中所提到的所有代码都可以到github中查看。项目场景简单的... 查看详情

[基于pytorch的mnist识别01]神经网络建立(代码片段)

写在前面前面我曾尝试在无框架的情况下进行神经网络的构建和调参，我发现虽然网络构建起来和运行起来都问题不大，但是在调参时就会显现无框架的弊端。经过初步的调参之后，我建立的网络识别准确率只能达到... 查看详情

[基于pytorch的mnist识别02]用户数据集的读取(代码片段)

写在前面pytorch包含了很多包括mnist在内的开源数据集，但是如果要建立自己的神经网络的话肯定需要训练自己的数据集，那么如何利用pytorch加载用户自己的数据集呢？今天就来解决这个问题。今天的工作需要加载用... 查看详情

图像分类基于pytorch搭建lstm实现mnist手写数字体识别（双向lstm，附完整代码和数据集）(代码片段)

...sdn.net/AugustMe/article/details/128969138文章中，我们使用了基于PyTorch搭建LSTM实现MNIST手写数字体识别，LSTM是单向的，现在我们使用双向LSTM试一试效果，和之前的单向LSTM模型稍微有差别，请注意查看代码的变化。1.导入依赖库这些依赖... 查看详情

用pytorch构建基于卷积神经网络的手写数字识别模型(代码片段)

本文参加新星计划人工智能(Pytorch)赛道：https://bbs.csdn.net/topics/613989052目录一、MINST数据集介绍与分析二、卷积神经网络三、基于卷积神经网络的手写数字识别一、MINST数据集介绍与分析 MINST数据库是机器学习领域... 查看详情

[pytorch系列-40]：卷积神经网络-模型的恢复/加载-搭建lenet-5网络与mnist数据集手写数字识别(代码片段)

作者主页(文火冰糖的硅基工坊)：文火冰糖（王文兵）的博客_文火冰糖的硅基工坊_CSDN博客本文网址：https://blog.csdn.net/HiWangWenBing/article/details/121132377目录第1章模型的恢复与加载1.1概述 1.2 模型的恢复与加载类型1.... 查看详情

[mnist06]pytorch(代码片段)

....在多种模型框架之间徘徊了一段时间之后,我最终选择了pytorch,因为他能兼容numpy,使用起来也较为简单,那么这篇文章我就主要分享pytorch环境的安装以及联合GPU的调试.主要内容主要内容在我的古月居博客&# 查看详情

基于pytorch平台实现对mnist数据集的分类分析（前馈神经网络softmax）基础版(代码片段)

基于pytorch平台实现对MNIST数据集的分类分析（前馈神经网络、softmax）基础版文章目录基于pytorch平台实现对MNIST数据集的分类分析（前馈神经网络、softmax）基础版前言一、基于“前馈神经网络”模型，分类分析... 查看详情

[pytorch系列-41]：卷积神经网络-模型参数的恢复/加载-搭建lenet-5网络与mnist数据集手写数字识别(代码片段)

作者主页(文火冰糖的硅基工坊)：文火冰糖（王文兵）的博客_文火冰糖的硅基工坊_CSDN博客本文网址：https://blog.csdn.net/HiWangWenBing/article/details/121183889目录第1章模型的恢复与加载1.1概述 1.2 模型的恢复与加载类型1.... 查看详情

小白学习pytorch教程十一基于mnist数据集训练第一个生成性对抗网络(代码片段)

@Author：RunsenGAN是使用两个神经网络模型训练的生成模型。一种模型称为生成网络模型，它学习生成新的似是而非的样本。另一个模型被称为判别网络，它学习区分生成的例子和真实的例子。生成性对抗网络2014࿰... 查看详情

小白学习pytorch教程十一基于mnist数据集训练第一个生成性对抗网络(代码片段)

图像分类基于pytorch搭建lstm实现mnist手写数字体识别（单向lstm，附完整代码和数据集）(代码片段)

写在前面：首先感谢兄弟们的关注和订阅，让我有创作的动力，在创作过程我会尽最大能力，保证作品的质量，如果有问题，可以私信我，让我们携手共进，共创辉煌。提起LSTM大家第一反应是在NLP的数据集上比较常见，不过在... 查看详情

pytorch学习(代码片段)

Pytorch1_手写字识别数据集的载入数据集展示网络搭建模型训练训练损失函数图模型预测下面将构造一个三层全连接层的神经网络来对手写数字进行识别。数据集的载入importtorchfromtorchimportnnfromtorch.nnimportfunctionalasFfromtorchimportoptimim... 查看详情

ai常用框架和工具丨13.pytorch实现基于cnn的手写数字识别

代码实例，PyTorch实现基于CNN的手写数字识别，希望对您有所帮助。文章目录环境说明一、模型训练1.1导入相关依赖1.2选择使用的硬件1.3超参数配置1.4准备训练集和测试集1.5模型创建1.6模型评估指标1.7模型训练1.8模型测试1.9模型... 查看详情

ai常用框架和工具丨13.pytorch实现基于cnn的手写数字识别

pytorch使用mnist数据集实现手写数字识别(代码片段)

...用mnist数据集实现手写数字识别是入门必做吧。这里使用pyTorch框架进行简单神经网络的搭建。首先导入需要的包。1importtorch2importtorch.nnasnn3importtorch.utils.dataasData4importtorchvision 接下来需要下载mnist数据集。我们创建train_data。使... 查看详情