pytorch
文章平均质量分 69
同志啊为人民服务!
我的快乐我做主!
展开
-
PyTorch框架中使用早停止Early Stopping(含详细代码)
文章目录1.什么是早停止?为什么使用早停止?2.如何使用早停止?3. Refferences1.什么是早停止?为什么使用早停止?早停止(Early Stopping)是 当达到某种或某些条件时,认为模型已经收敛,结束模型训练,保存现有模型的一种手段。机器学习或深度学习中,有很大一批算法是依靠梯度下降,求来优化模型的。是通过更新参数,让Loss往小的方向走,来优化模型的。可参考BP神经网络推导过程详解关于模型何时收敛(模型训练好了,性能达到要求了或不能再优化了),此时我们可以采取一些判断标准:1.验原创 2022-03-21 21:46:49 · 38316 阅读 · 16 评论 -
TensorFlow和PyTorch框架绘制loss和accuracy曲线
文章目录1.前言2.TensorFlow框架绘制loss和accuracy曲线3.PyTorch框架绘制loss和accuracy曲线4.pytorch中使用tensorboard绘制Accuracy/Loss曲线1.前言搞深度学习时,我们一般会遇到一个问题,评估模型的性能:这时,一般会绘制模型在训练集 & 验证集上的准确率值和损失值曲线。在这一点上TensorFlow框架比PyTorch框架方便,因为TensorFlow框架在训练模型时,把模型、训练集和验证集上的accuracy和loss都原创 2022-03-17 23:16:36 · 21332 阅读 · 10 评论 -
神经网络模型训练时,性能不高?原因总结(神经网络模型性能优化方法汇总)
文章目录 前言1. 改模型框架2.数据处理3.超参数对模型训练的影响3.1 学习率对模型训练时的影响3.2 批处理大小对模型训练时的影响3.3 epoch对模型训练时的影响3.4 如何判断模型收敛3.5模型过拟合时怎么办?原创 2022-03-15 21:58:40 · 8988 阅读 · 2 评论 -
Hu不变矩+BP神经网络,实现对图像的分类(pytorch实现)
先对图片,用hu矩进行特征提取,每个图片产生7个特征值。这7个特征值作为数据,再加上一个标签值。然后送入BP神经网络中,进行模型的训练。模型训练好后,就可以使用模型,对一个新的图片,进行分类识别了。原创 2022-03-09 14:39:54 · 2555 阅读 · 42 评论 -
用残差网络实现MNIST数据集手写数字识别
1.残差网络本文为用带残差块的CNN网络实现MNIST数据集手写数字的识别。关于残差网络,知乎上有篇文章讲的不错,供参考:详解残差网络残差网络比起LeNet等简单的神经网络,不同之初在于,多了一个连接线。左边为基础的CNN结构,右边为带残差的网络结构残差块是目前网络模型中,一个跟经典、很基础的结构,像DenseNet就是基于残差块来提出的,一个新的网络模型。2.MNIST数据集参考笔者的上篇博客:CNN实现MNIST数据集手写数字识别3.模型结构Residual Block:残差块原创 2022-01-05 23:05:49 · 2355 阅读 · 4 评论 -
CNN实现MNIST数据集手写数字识别
1.MNIST数据集MNIST数据集是由0 到9 的手写数字图像构成的。训练图像有6 万张,测试图像有1 万张每一张图片都有对应的标签数字。因此这个测试集就可以作为验证集使用。MNIST的图像,每张图片是包含28 像素× 28 像素的灰度图像(1 通道),各个像素的取值在0 到255 之间。每张图片都由一个28 ×28 的矩阵表示,每张图片都由一个784 维的向量表示(28*28=784)。详细介绍参考:http://yann.lecun.com/exdb/mnist/2.CNN的基础卷积和池化原创 2022-01-04 23:05:25 · 7718 阅读 · 3 评论 -
神经网络做MNIST手写数字识别代码
代码(python+pytorch)import torchfrom torchvision import transformsfrom torchvision import datasetsfrom torch.utils.data import DataLoaderimport torch.optim as optimimport torch.nn.functional as Fimport matplotlib.pyplot as pltbatch_size = 64tran原创 2021-12-24 23:48:30 · 5846 阅读 · 0 评论 -
AttributeError: ‘builtin_function_or_method‘ object has no attribute ‘view‘解决办法
1.问题陈述今天在使用pytorch中的view方法,改变张量形状时,发生错误。解决办法检查forward在使用时,传入的参数x是什么类型,是否是张量,是否传参正确。检查后发现,不是张量,改为张量。问题解决。...原创 2021-12-24 22:49:38 · 5341 阅读 · 0 评论