nn.Module

pytorch中指定或者冻结一层网络的权值（weight)

一.指定权值如果有这样一个神经网络importtorch.nnasnnimporttorch.nn.functionalasFclassNetwork(nn.Module):def__init__(self
21岁害怕编程·2022-12-20 22:56

深度学习与物体检测每日笔记（5）Pytorch神经网络库torch.nn

nn.Module类损失函数优化器nn.optimnn.Module：nn.Module是pytorch提供的神经网络类，在该类中实现了网络各层定义及前向和反向传播。
胳膊·2022-12-20 19:29

Pytorch学习率lr衰减(decay)(scheduler)(一)——(LambdaLR&StepLR&MultiStepLR&ExponentialLR&ReduceLROnPlateau )

1、手动修改optimizer中的lrimportmatplotlib.pyplotaspltfromtorchimportnnimporttorchclassNet(nn.Module):def__init
hxxjxw·2022-12-20 18:37

mvit代码

多头注意力机制classMultiHeadedAttention(nn.Module):def__init__(self,h,d_model,dropout=0.1):"Takeinmodelsizeandnumberofheads
你饿了嘛？？·2022-12-20 15:42

conv2d() received an invalid combination of arguments问题解决

classSynthesizedImage(nn.Module):def__in
iwill323·2022-12-20 15:39

pytorch 使用LSTM进行手写数字识别

模型的搭建importtorchimporttorch.nnasnnclassRNN(nn.Module):def__init__(self):super(RNN,self).
南妮儿·2022-12-20 11:29

Yolov5更换BiFPN的详细步骤总结

第一步：修改common.py将如下代码添加到common.py文件中#BiFPN#两个特征图add操作classBiFPN_Add2(nn.Module):def__init__(self,c1,c2
·2022-12-20 06:51

【上采样】——深度学习上采样方式

transposedconvolutions光流上采样SFNetlearnedupsamplehttps://arxiv.org/pdf/2011.06961.pdfcodeclassUpsample(nn.Module
农夫山泉2号·2022-12-20 04:06

pytorch反卷积函数ConvTransposed2d参数以及使用方法

ConvTransposed2d参数使用方法1.打五角星的5个公式，只要满足前1，2，3，5，kernal_size和stride大小随便选2.中间运算流程看看就行了，用于了解每个参数的意义classBasicDeConv2d(nn.Module
qq_40840829·2022-12-19 17:47

Pytorch中的卷积操作——nn.Conv2d, nn.ConvTranspose2d

前言pytorch中用nn.Module实现卷积操作总共有3种Conv1d,Conv2d,Conv3d而这三种方式只是卷积核在不同的维度上进行卷积的结果，基本参数一致，这里以Conv2d为例进行介绍。
菜鸟12134·2022-12-19 17:16

ResNeSt的代码阅读（pytorch版本）ResNeSt Split-Attention Networks

ResNeSt、split-attention和ablationsplit-attention模块也就是下面的图：上图对应论文中的：ResNet就是基本的resnet的编码：classBottleneck(nn.Module
佑林杉·2022-12-19 11:28

PyTorch实现Vision Transformer

importtorchfromtorchimportnn,einsumimporttorch.nn.functionalasFfromeinopsimportrearrange,repeatclassResidual(nn.Module
FPGA硅农·2022-12-19 11:43

Attention和Transformer的Pytorch实现

importtorchimporttorch.nnasnnclassAttentionBlock(nn.Module):def__init__(self,input_dim,hidden_dim,num_heads
disanda·2022-12-19 11:12

TorchScript (将动态图转为静态图)(模型部署)(jit)(torch.jit.trace)(torch.jit.script)

TorchScript是Pytorch模型（继承自nn.Module）的中间表示，保存后的torchscript模型可以在像C++这种高性能的环境中运行TorchScript是一种从PyTorch代码创建可序列化和可优化模型的方法
hxxjxw·2022-12-19 11:03

PyTorch学习笔记(3)

2不用PyTorch框架特性搭建网络2.1MNIST数据集的建立2.2不使用torch.nn从零建立神经网络3开始用PyTorch特性模块替换3.1使用torch.nn函数3.2使用nn.Module模块
CarnivoreRabbit·2022-12-19 00:29

PyTorch Tutorials 摘要(3) What is torch.nn really?

文章目录前言源码前言教程链接概述：看完之后大概能明白nn.Module/nn.Parameter/torch.optim/Dataset/DatasetLoader的作用。
清欢守护者·2022-12-19 00:29

关于pytorch官网教程中的What is torch.nn really?（二）

文章目录Using`torch.nn.functional`Refactorusing`nn.Module`Refactorusing`nn.Linear`Refactorusing`optim`Refactorusing
MYTCHITOS·2022-12-18 23:51

循环神经网络（RNN、LSTM、GRU）的结构及代码实现

RNN代码实现：#一般循环神经网络RNNclassConvRNN(nn.Module):def__init__(self,inp_dim,oup_dim,kernel,dilation):super()
乐亦亦乐·2022-12-18 15:13

pytorch实现resnet36并且用tensorboard可视化

from.BasicModuleimportBasicModulefromtorchimportnnimporttorchfromtorch.nnimportfunctionalasFimporttorchfromtensorboardXimportSummaryWriterimporttorch.onnxclassResidualBlock(nn.Module
cncxz5801·2022-12-18 14:25

Pytorch框架 || torch.nn.modules.Module(nn.Module)

文章目录1一个简单的网络2nn.Module.init_weight()2.1kaiming高斯初始化1一个简单的网络一个Pytorch模型应该以类的形式出现Pytorch训练模型应该是nn.Module
谜底是你_·2022-12-18 13:47

『PyTorch』第十三弹_torch.nn.init参数初始化

初始化参数的方法nn.Module模块对于参数进行了内置的较为合理的初始化方式，当我们使用nn.Parameter时，初始化就很重要，而且我们也可以指定代替内置初始化的方式对nn.Module模块进行补充
weixin_33850890·2022-12-18 13:45

pytorch教程之nn.Module类详解——使用Module类来自定义网络层

前言：前面介绍了如何自定义一个模型——通过继承nn.Module类来实现，在__init__构造函数中申明各个层的定义，在forward中实现层之间的连接关系，实际上就是前向传播的过程。
三世·2022-12-18 13:42

pytorch实现segnet_SegNet网络的Pytorch实现

1importtorch.nnasnn2importtorch34classconv2DBatchNormRelu(nn.Module):5def__init__(self,in_channels,out_channels
小女巫诺诺·2022-12-18 11:58

8 - nn.Module源码讲解(下)

文章目录1.模型的保存和加载(权重保存-简易方法)2.保存checkpoint2.1导入所有相关的库2.2定义并初始化神经网络2.3初始化优化器2.4保存常规检查点checkpoint2.5加载常规检查点checkpoint3.to操作4._parameters和parameters的区别5.state_dict6.parameters&named_parameters7._modules&nam
取个名字真难呐·2022-12-18 02:24

day25 torch.nn 操作2 入门 一起学 PyTorch吧，PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习，共同进步 坚持

importtorch.nnasnnclassModel(nn.Module):def__init__(self):super(Model,self).
Cchaofan·2022-12-18 00:33

YOLOv5 ——极市平台：口罩识别

yolov56.1版本的模型输出包含了输入的部分，需要去掉（最新版本代码添加了一个self.export判断，去掉了export时输出x问题），输出只需要output这一项：#yolo.pyclassDetect(nn.Module
ZerosCoder·2022-12-17 18:11

询问ChatGPT如何使用pytorch构建CNN卷积神经网络

ChatGPT的回答如下下面是一个使用PyTorch构建卷积神经网络（CNN）的示例代码：importtorchimporttorch.nnasnnclassCNN(nn.Module):def__init
hellangle4742·2022-12-17 15:35

pytorch模型初始化参数

这个一般在构建模型nn.Module类时就完成比较容易。
白三点·2022-12-17 11:04

profile计算模型参数

fromthopimportprofileclassTest(nn.Module):def__init__(self,input_size,output_szie):super(Test,self).
Douzi1024·2022-12-17 08:16

自己搭建Resnet50并在FashionMNIST训练-pytorch

importtorchimporttorch.nnasnnfromtorch.nnimportfunctionalasFimporttorchsummaryclassBottleneck(nn.Module
ZzBoAYU·2022-12-16 17:52

4.pytorch学习：pool2d——2d池化

理解池化加入图像数据集自建一个tensor理解池化最大池化nn.MaxPool2dimporttorchfromtorchimportnninput=torch.randn(1,3,5,5)classZiDingYi(nn.Module
机械专业的计算机小白·2022-12-16 12:41

Bert+BiLSTM+attention+TextCNN来实现文本分类

importtorchimporttorch.nnasnnimporttorch.nn.functionalasFimporttorch.optimasoptimfromtorchtext.datasetsimporttext_classificationfromtorchtext.dataimportField,BucketIteratorclassTextCNN(nn.Module
Makup·2022-12-16 12:45

Bert+Bilstm+attention模型

importtorchimporttorch.nnasnnclassBERTLSTMAttention(nn.Module):def__init__(self,bert_model,hidden_size
Makup·2022-12-16 12:45

基于pytorch的自注意力机制实现

基于pytorch的自注意力机制实现importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassNet(nn.Module):def__init
DreamBoy@·2022-12-16 12:39

Focus的作用及pytorch实现

用于切片操作，比如4*4*3变成了2*2*12的特征图classFocus(nn.Module):#Focuswhinformationintoc-spacedef__init__(self,c1,c2
Alphapeople·2022-12-16 11:26

yolov5 focus

kernel,padding#Padto'same'ifpisNone:p=k//2ifisinstance(k,int)else[x//2forxink]#auto-padreturnpclassConv(nn.Module
weixin_43489941·2022-12-16 11:26

Pytorch网络结构可视化：torchsummary

importtorchimporttorch.nnasnnfromtorchsummaryimportsummaryclassNetwork(nn.Module):def__init__(self,channels_img
u013250861·2022-12-16 10:31

【深度学习】可视化pytorch网络结构

1.构建你的pytorch网络fromComponentsimport*classPortraitNet(nn.Module):def__init__(self,num_classes=2):super
Baily24·2022-12-16 10:00

pytorch用于多标签分类的bceloss

F.logsigmoid(-output)loss=-positive_prob*target-negative_prob*(1-target)loss=loss.mean()returnlossclassbceloss(nn.Module
小星星么么哒·2022-12-16 08:28

AttributeError: Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from ‘D:\\ModelTest\\yolov5-5

module'models.common'from'D:\\ModelTest\\yolov5-5.修改一下那个models文件夹下的common.py添加下面代码即可importwarningsclassSPPF(nn.Module
Ricky-T·2022-12-15 18:12

AttributeError: Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from ‘model\\common.py‘

解决方案找到common.py文件，在里面添加如下代码：importwarningsclassSPPF(nn.Module):#SpatialPyramidPooling-Fast(SPPF)layerforYOLOv5byGlennJocherdef
虚拟指尖·2022-12-15 18:37

运行yolov5报错 AttributeError: Cant get attribute SPPF on module models.common

原因在models/common.py里面找不到类SPPF2.解决方案：粘贴以下内容到models/common.py注意：先importwarningsimportwarningsclassSPPF(nn.Module
Newjet666·2022-12-15 18:04

YOLOv5训练报错：Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘解决方案

在model/common.py这个文件里添加以下代码：importwarningsclassSPPF(nn.Module):#SpatialPyramidPooling-Fast(SPPF)layerforYOLOv5byGlennJocherdef
加斯顿工程师·2022-12-15 18:03

【Debug记录】跑YOLOv5时报错AttributeError: Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from ‘D:\\P

问题：AttributeError:Can'tgetattribute'SPPF'on解决方法：在models/common.py中添加类SPPF，代码如下：classSPPF(nn.Module):def
姚路遥遥·2022-12-15 18:00

pytorch搭建RNN-LSTM循环神经网络[回归]详解

可参考pytorch搭建CNN卷积神经网络详解搭建RNN（该任务使用RNN足矣）classRNN(nn.Module):def__init__(self):super(RNN,self).
troublemaker、·2022-12-15 15:00

pytorch06

CIFAR10数据集Lenet5importtorchfromtorchimportnnfromtorch.nnimportfunctionalasFclassLenet5(nn.Module):"""
非畅6 1·2022-12-15 14:46

Pytorch中nn.Module中的self.register_buffer解释

self.register_buffer作用解释今天遇到了这样一种用法，self.register_buffer(‘name’,Tensor)，该方法的作用在于定义一组参数。该组参数在模型训练时不会更新（即调用optimizer.step()后该组参数不会变化，只可人为地改变它们的值），但是该组参数又作为模型参数不可或缺的一部分。实验四种方式初始化模型中的参数定义常见模型时的操作使用registe
丶夜未央丶·2022-12-14 20:11

PyTorch 2.0 实操：为 HuggingFace 和 TIMM 模型提速！

它作为torch.jit.script()的直接替代品，可以直接在nn.Module上运行，无需修改源代码。上篇文章中，我们介绍了torch.compile支持任意的
·2022-12-14 20:42

神经网络复现步骤

一、神经网络1、根据architecture图，记录通道数变化引入importtorch.nnasnn创建通用卷积块方法2、定义类，继承(nn.Module):de__init__中初始化每个卷积块或设定的结构块注意
JYliangliang·2022-12-14 18:49

53、正确导出onnx模型

基本思想：学习tensorRT教程，来自bilibi,参考附录一一、代码：importtorchimporttorch.nnasnnclassModel(nn.Module):def__init__(self
sxj731533730·2022-12-14 16:33
上一页 11 12 13 14 15 16 17 18 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他