nn.Module

transformer模型构建

接下来就来实现完整的编码器-解码器结构.Transformer总体架构图:编码器-解码器结构的代码实现#使用EncoderDecoder类来实现编码器-解码器结构classEncoderDecoder(nn.Module
AI耽误的大厨·2025-02-20 16:03

yolov5 pt-＞onnx-＞om yolov5模型转onnx转om模型转换

yolov5pt->onnx->omyolov5-6.1版本models/yolo.pyDetect函数修改classDetect(nn.Module):defforward(self,x):z=[]#
qq_43650438·2025-02-19 05:14

大模型笔记：pytorch实现MOE

0导入库importtorchimporttorch.nnasnnimporttorch.nn.functionalasF1专家模型#一个简单的专家模型，可以是任何神经网络架构classExpert(nn.Module
UQI-LIUWJ·2025-02-12 23:37

一文解释nn、nn.Module与nn.functional的用法与区别

个人主页：十二月的猫-CSDN博客系列专栏：零基础入门PyTorch框架_十二月的猫的博客-CSDN博客十二月的寒冬阻挡不了春天的脚步，十二点的黑夜遮蔽不住黎明的曙光目录1.前言2.Torch.nn库3.nn.Module4.nn.functional4.1基本用法4.2常用的functional中的函数4.2.1激活函数4.2.2损失函数4.2.3非线性操作5.小例子6.总结1.前言《零基础入门
十二月的猫·2025-02-10 03:23

解决TypeError: __init__() takes 1 positional argument but 2 were given

问题描述：如下图，在使用torch.nn.Sigmoid非线性激活时报错源代码：classtestrelu(nn.Module):def__init__(self):super().
求不脱发·2025-02-10 01:07

从零学习pytorch——4.基础概念讲解及神经网络模块

从零学习pytorch——4.基础概念讲解及神经网络模块在PyTorch中，nn.Module是构建神经网络模型的基础类。
PyTorch 研习社·2025-02-07 22:05

torch库介绍

文章目录基本概念张量（Tensor）自动求导（AutomaticDifferentiation）：神经网络模块（nn.Module）优化器（Optimizer）：导入torch示例构建神经网络总结torch
codelyq·2025-02-07 15:22

自定义数据集 使用pytorch框架实现逻辑回归并保存模型，然后保存模型后再加载模型进行预测，对预测结果计算精确度和召回率及F1分数

定义逻辑回归模型：继承nn.Module类，定义一个线性层，并在forward方法中应用sigmoid激活函数。训练模型：使用二元交叉熵损失函数BCELoss和随机梯度下降优化器SGD进行训练。
知识鱼丸·2025-02-05 06:42

YOLOv10-1.1部分代码阅读笔记-tasks.py

tasks.pyultralytics\nn\tasks.py目录tasks.py1.所需的库和模块2.classBaseModel(nn.Module):3.classDetectionModel(BaseModel
红色的山茶花·2025-02-01 03:27

YOLOv10-1.1部分代码阅读笔记-loss.py

loss.pyultralytics\utils\loss.py目录loss.py1.所需的库和模块2.classVarifocalLoss(nn.Module):3.classFocalLoss(nn.Module
红色的山茶花·2025-02-01 03:27

YOLOv9-0.1部分代码阅读笔记-common.py

common.pymodels\common.py目录common.py1.所需的库和模块2.defautopad(k,p=None,d=1):3.classConv(nn.Module):4.classAConv
红色的山茶花·2025-02-01 03:26

YOLOv7-0.1部分代码阅读笔记-common.py

common.pymodels\common.py目录common.py1.所需的库和模块2.defautopad(k,p=None):3.classMP(nn.Module):4.classSP(nn.Module
红色的山茶花·2025-02-01 03:26

YOLOv8-ultralytics-8.2.103部分代码阅读笔记-block.py

block.pyultralytics\nn\modules\block.py目录block.py1.所需的库和模块2.classDFL(nn.Module):3.classProto(nn.Module
红色的山茶花·2025-02-01 03:26

Transformer模型解码器部分实现

代码实现：classDecoderLayer(nn.Module):def__init__(self,size,self_attn,src_attn,feed_forward,
APPLECHARLOTTE·2025-02-01 03:24

YOLOv11-ultralytics-8.3.67部分代码阅读笔记-head.py

head.pyultralytics\nn\modules\head.py目录head.py1.所需的库和模块2.classDetect(nn.Module):3.classSegment(Detect
红色的山茶花·2025-01-30 19:53

YOLOv11-ultralytics-8.3.67部分代码阅读笔记-transformer.py

transformer.pyultralytics\nn\modules\transformer.py目录transformer.py1.所需的库和模块2.classTransformerEncoderLayer(nn.Module
红色的山茶花·2025-01-28 08:47

pytorch 手写数字识别

importnumpyasnpimporttorchimporttorch.nnasnnimporttorch.nn.functionalasFimportstructimporttorch.optimasoptimfromPILimportImagefrommatplotlibimportpyplotaspltclassNet(nn.Module
CrxzYia·2025-01-27 20:54

YOLOv11-ultralytics-8.3.67部分代码阅读笔记-conv.py

conv.pyultralytics\nn\modules\conv.py目录conv.py1.所需的库和模块2.defautopad(k,p=None,d=1):3.classConv(nn.Module
红色的山茶花·2025-01-26 14:24

YOLOv10-1.1部分代码阅读笔记-model.py

model.pyultralytics\engine\model.py目录model.py1.所需的库和模块2.classModel(nn.Module):1.所需的库和模块#UltralyticsYOLO
红色的山茶花·2025-01-20 20:57

深度学习模块C2f代码详解

下面是对C2f类的详细解释：类定义和初始化Python复制classC2f(nn.Module):“”“FasterImplementationofCSPBottleneckw
你是狒狒吗·2025-01-18 12:44

Pytorch实现猫狗分类

在进行编码之前，确保已经安装好Pytorch以及相关依赖环境模型定义构建构建一个较为简单的卷积神经网络模型#定义神经网络模型classCatDogClassifier(nn.Module):def__init
NoKnowovo·2025-01-17 13:01

【pytorch】register_buffer的使用

下面列出不需要手动注册、会自动注册的几种情况：1.nn.Parameter自动注册：任何你在nn.Module中定义为nn.Parameter的张量都会自动注册为模型的参数。
Aha_aho·2024-09-10 18:12

神经网络的基本骨架——nn.Module的使用

这是官网给出的一个示例，首先继承父类nn.Module并初始化，然后重载前向传播foward方法。
搁浅丶.·2024-09-08 12:27

LLM手撕

LayerNormimporttorchfromtorchimportnnclassLayerNorm(nn.Module):def__init__(self,hidden_size,eps=1e-6)
比起村村长·2024-09-04 04:26

pytorch模型保存及加载参数恢复训练的例子

优化器的状态以恢复训练：保存与加载模型和优化器状态的示例importtorchimporttorch.nnasnnimporttorch.optimasoptim#定义一个简单的模型classSimpleModel(nn.Module
静心问道·2024-09-03 21:44

3.关于Detr

关于Detr模型架构总体架构classTransformer(nn.Module):def__init__(self,d_model=512,nhead=8,num_encoder_layers=6,num_decoder_layers
安逸sgr·2024-09-03 19:05

Transformer模型整体构建的实现

编码器-解码器结构classEncoderDecoder(nn.Module):def__init__(self,encoder,decoder,source_embedding,target_embedding
好好学习Py·2024-08-30 22:14

神经网络分类任务python入门

torch.nn.functional创建一个model来更简化代码使用TensorDataset和DataLoader来简化整个过程Mnist分类任务-网络基本构建与训练方法，常用函数解析-torch.nn.functional模块-nn.Module
三十度角阳光的问候·2024-08-30 05:24

Pytorch构建网络模型结构都有哪些方式

目录前言1.使用nn.Module基类2.使用nn.Sequential容器3.使用nn.ModuleList4.使用nn.ModuleDict5.混合使用nn.Module和原生Python代码6.表格总结前言
松小白song·2024-08-26 16:57

PyTorch踩坑记录1

网上查到是因为model.eval()会影响继承nn.module类的
普通攻击往后拉·2024-08-23 14:43

Pytorch nn.Module

一、torch.nn简介torch.nn是PyTorch中用于构建神经网络的模块。它提供了一系列的类和函数，用于定义神经网络的各种层、损失函数、优化器等。torch.nn提供的类：Module:所有神经网络模型的基类，用于定义自定义神经网络模型。Linear:线性层，进行线性变换。Conv2d:二维卷积层。RNN,LSTM,GRU:循环神经网络层，分别对应简单RNN、长短时记忆网络（LSTM）、门
霖大侠·2024-03-16 15:53

Pytorch关于CIFAR-10测试

importtorchfromtorchimportnnfromtorch.nnimportConv2d,MaxPool2d,Flatten,Linear,Sequential#定义网络结构classModel(nn.Module
月疯·2024-02-20 07:03

pytorch神经网络入门代码

importtorchimporttorch.nnasnnimporttorch.optimasoptimimporttorchvisionimporttorchvision.transformsastransforms#定义神经网络结构classSimpleNN(nn.Module
hehui0921·2024-02-20 00:15

pytorch创建模型方式

1.继承自nn.Module的方式fromtorchimportnnimporttorch.nn.functionalasF'''继承自nn.Module'''classLModel(nn.Module
守护安静星空·2024-02-19 21:43

pytorch——保存‘类别名与类别数量’到权值文件中

在自己的类中加入想要加入数据信息classyour_nets(nn.Module):def__init__(self,num_classes=
黑夜寻白天·2024-02-06 17:02

20天吃pytorch 1-2 图片数据建模流程遇到问题及分析

学习内容：1-2图片数据建模流程如下1、datasets.ImageFolder图片数据加载，然后送入Dataloader做一个数据管道2、nn.Module基类构建自定义模型3、模型训练步骤4、模型训练评估遇到问题与解决方案
七月的和弦·2024-02-06 11:29

Transformer实战-系列教程5：Vision Transformer 源码解读3

源码解读1VisionTransformer源码解读2VisionTransformer源码解读3VisionTransformer源码解读46、Block类------构造函数classBlock(nn.Module
机器学习杨卓越·2024-02-06 05:45

（超详细）9-YOLOV5改进-添加EffectiveSEModule注意力机制

importtorchfromtorchimportnnasnnfromtimm.models.layers.create_actimportcreate_act_layerclassEffectiveSEModule(nn.Module
我要变胖哇·2024-02-06 03:14

（超详细）8-YOLOV5改进-添加EMA意力机制

1、在yolov5/models下面新建一个EMA.py文件，在里面放入下面的代码代码如下：importtorchfromtorchimportnnclassEMA(nn.Module):def__init
我要变胖哇·2024-02-06 03:43

PyTorch 2.2 中文官方教程（十四）

这样做就像编写自己的nn.Module一样容易。对深度学习模型进行正则化是一项令人惊讶的挑战。传统技术，如惩罚方法，通常在应用于深度模型时效果不佳，因为被优化的函数的复杂性。
绝不原创的飞龙·2024-02-04 21:55

【深度学习】基于PyTorch架构神经网络学习总结（基础概念&基本网络搭建）

nn.Module的使用利用PyTorch架构使用神经网络模型时，一般是利用torch.nn函数自定义神经网络框架|官方示例：importtorch.nnasnnimporttorch.nn.functionalasFclassModel
hi_ly_51·2024-02-02 13:25

PyTorch 中神经网络库torch.nn的详细介绍

以下是torch.nn的关键组成部分及其功能：nn.Module类：nn.Module是所有自定义神经网络模型的基类。
科学禅道·2024-02-02 09:23

【极简】Pytorch中的register_buffer()

使用：只要是nn.Module的子类就能直接self.调用使用：classA(nn.Module):#...self.register_buffer('betas',torch.linspace(beta
Yonggie·2024-02-02 08:46

Bi-Lstm+crf命名实体识别任务中crf的作用

这是一段使用百度ernie-1.0做特征提取的Bi-Lstm+crf的代码：classERNIE_LSTM_CRF(nn.Module):"""ernie_lstm_crfmodel"""def__init
sunshine2853·2024-02-01 21:25

Twin Contrastive Learning for Online Clustering

arxiv.org/pdf/2210.11680.pdf这篇文章利用双路对比学习实现了在线聚类，对我这个方向有一定帮助，以下是此双路对比学习的对比损失函数实例对比损失classInstanceLoss(nn.Module
Mr___WQ·2024-02-01 15:12

transformer_多头注意力机制代码笔记

transformer_多头注意力机制代码笔记以GPT-2中多头注意力机制代码为例classCausalSelfAttention(nn.Module):"""因果掩码+多头自注意力机制Avanillamulti-headmaskedself-attentionlayerwithaprojectionattheend.Itispossibletousetorch.nn.MultiheadAtten
等风来随风飘·2024-02-01 08:29

使用libtorch实现sobel算子进行边缘检测，python和c++实现

pythonimporttorchimporttorch.nnasnnfromPILimportImageimportnumpyasnpimportcv2classSobelEdgeDetector(nn.Module
吴天德少侠·2024-01-31 16:20

机器学习复习（2）——线性回归&SGD优化算法

目录线性回归代码线性回归理论线性回归代码classMy_Model(nn.Module):def__init__(self,input_dim):super(My_Model,self).
不会写代码！！·2024-01-31 07:38

类参数和实例参数

classConv2D(nn.Module):def__init__(self,kernel_size):super().
代码猪猪傻瓜coding·2024-01-31 02:57

PyTorch动态计算图构建的详细过程和代码

当用户定义了一个继承自nn.Module的神经网络模型，并通过调用model.forward(input)进行前向传播时，PyTorch会根据执行的张量操作序列自动构建并维护一个动态计算图，其中的详细过程是
科学禅道·2024-01-30 18:42
上一页 1 2 3 4 5 6 7 8 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他