nn.Module

【深度学习Week4】MobileNet_ShuffleNet

cryptography==36.0.2版本，以及更换下载链接的方法，都不行，最后采用了手动下载mat文件并上传到colab的方法高光谱图像分类数据集简介IndianPines&mat定义网络：classHybridSN(nn.Module
跪求指点·2023-08-09 03:19

PyTorch Lightning基础入门

PyTorchLightningpipinstalllightningcondainstalllightning-cconda-forge定义一个LightningModuleLightningModule可以让pytorch的nn.Module
HMTT·2023-08-08 14:11

获得模型中某一层特征图的尺寸大小

以下是一个示例代码，展示如何获得不同层的特征图尺寸：importtorchimporttorch.nnasnnclassYourModel(nn.Module):def__init__(self):super
Make_magic·2023-08-07 23:21

pytorch存储模型

只存储模型参数importtorchimporttorch.nnasnn#定义一个简单的模型classMyModel(nn.Module):def__init__(self):super(MyModel
希望自己能一直努力·2023-08-07 13:43

TorchScript 的简介，TorchScript 是 PyTorch 模型(`nn.Module`的子类）的中间表示形式，可以在高性能环境(例如 C ++）中运行

TorchScript简介本教程是TorchScript的简介，TorchScript是PyTorch模型(nn.Module的子类）的中间表示形式，可以在高性能环境(例如C++）中运行。
yanglamei1962·2023-08-07 11:09

Pytorch nn.Parameter() (register_parameter)

torch.nn.Parameter是继承自torch.Tensor的子类，其主要作用是作为nn.Module中的可训练参数使用。
hxxjxw·2023-08-06 16:13

Pytorch中nn. 与nn.functional 的区别

先上一篇我觉得写得非常清楚的博客，详细内容见下https://blog.csdn.net/chanbo8205/article/details/114890006简要概略：nn.XXX均为一个类的实现，其均继承自一个祖父类nn.Module
Petersburg·2023-08-06 07:34

PyTorch中nn-XXX与F-XXX的区别

它们的主要区别如下：nn中的模块是以类形式存在的；F中的模块是以函数形式存在的nn中的模块是nn.Module的子类，包含可学习参数、可导，在反向传播中可以计算梯度，可以在模型中作为
header-files·2023-08-06 07:03

6.2 手写卷积类

importtorchfromtorchimportnnfromd2limporttorchasd2lclassConv2D(nn.Module):def__init__(self,kernel_size
computer_vision_chen·2023-08-06 07:15

深度学习调参入门

深度学习调参入门1、搭建ANN模型2、lr_scheduler学习率参数管理3、对训练数据进行标准化和反标准化1、搭建ANN模型在python中，使用nn.Module和nn.Sequential都可以用来定义神经网络
ZEERO~·2023-08-06 06:27

根据Pytorch源码实现的 ResNet18

一，类模块定义:importtorchimporttorch.nnasnnimporttorch.nn.functionalasFfromtorchimportTensorclassResBlock(nn.Module
gwd777·2023-08-05 06:57

自然语言文本分类模型代码

的文本分类模型的示例代码，用于将给定的文本分为多个预定义类别：importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassTextClassifier(nn.Module
Dream SYC_UCC·2023-08-04 14:06

【torch保存神经网络模型并加载和测试】

加载的方式如下：classMLP(nn.Module):def__init__(self,n_in,n_hid1,n_hid2,n_out):su
知识推荐号·2023-08-02 17:48

注意力机制代码

importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassAttn(nn.Module):def__init__(self,query_size
a2428083131·2023-08-02 12:10

用pytorch写一段代码：用CNN网络对电机轴承的一维振动信号做故障诊断，包括：正常、滚动体故障、外圈故障、内圈故障共四类状态，实现给定真实振动信号判别故障类别...

好的，以下是使用PyTorch实现的一维振动信号故障诊断代码：importtorchimporttorch.nnasnnimporttorch.optimasoptim#定义CNN模型classNet(nn.Module
轮胎技术Tyretek·2023-08-01 00:09

深度学习中简易FC和CNN搭建

人工智能研究院（FacebookAIResearch）开发的Torch和cuda版本的对应，手动安装较好全连接FC(Batch*Num)搭建建议网络：fromtorchimportnnclassMnist_NN(nn.Module
钟钟终·2023-07-31 13:55

super(Module_ModuleList, self).__init__()的作用是什么？

classModule_ModuleList(nn.Module):def__init__(self):super(Module_ModuleList,self).
有梦想的咕噜·2023-07-31 09:53

Pytorch(二)

一、分类任务构建分类网络模型必须继承nn.Module且在其构造函数中需调用nn.Module的构造函数无需写反向传播函数，nn.Module能够利用autograd自动实现反向传播Module中的可学习参数可以通过
几两春秋梦_·2023-07-31 07:38

Backbone共享参数，代码出现的一个错误

A-deeply-supervised-image-fusion-network-for-change-detection-in-remote-sensing-images/blob/master/pytorch%20version/DSIFN.py这块是共享一个backbone特征提取网络classDSIFN(nn.Module
_LvP·2023-07-31 06:39

pytorch实现一个简单的CNN

importtorchimporttorch.nnasnnimporttorch.optimasoptimimporttorchvisionimporttorchvision.transformsastransforms#DefinetheCNNmodelclassCNN(nn.Module
壹晴天·2023-07-30 21:01

pytorch（6）——神经网络基本骨架nn.module的使用

1神经网络框架1.1Module类的使用NN(Neuralnetwork):神经网络Containers:容器ConvolutionLayers:卷积层Poolinglayers:池化层PaddingLayers:填充层Non-linearActivations(weightedsum,nonlinearity):非线性激活Non-linearActivations(other):非线性激活Nor
阳排·2023-07-30 06:23

transformer代码注解

classPositionWiseFFN(nn.Module):'''ffn_num_inputs4ffn_num_hiddens4ffn_num_outputs8'''def__init__(self
the animal·2023-07-29 14:01

3D点云基本网络模块（一）：Spatial Transformer Networks（STN）

fromtorch.autogradimportVariableimportutilsclassSTN(nn.Module):def__init__(self,num_scales=1,num_points
Y.K.Hou·2023-07-29 12:27

PyTorch笔记--3 torch中nn.relu和f.relu的区别

torch.nn.Reluandtorch.nn.functional.relu结果一致，不同点如下1、nn.Relu不同的是nn.Relu作为一层结构，必须添加到nn.Module容器中才能使用使用方式如下
梦在黎明破晓时啊·2023-07-29 08:21

Pytorch深度学习-----神经网络的基本骨架-nn.Module的使用

系列文章目录PyTorch深度学习——Anaconda和PyTorch安装Pytorch深度学习-----数据模块Dataset类Pytorch深度学习------TensorBoard的使用Pytorch深度学习------Torchvision中Transforms的使用（ToTensor，Normalize，Resize，Compose，RandomCrop）Pytorch深度学习-----
-希冀-·2023-07-28 13:54

K折交叉验证

importtorchimporttorch.nnasnnfromsklearn.model_selectionimportKFoldfromsklearn.metricsimportaccuracy_score#定义一维卷积网络模型classConvNet(nn.Module
灰太狼家的小鸭子·2023-07-28 12:55

yolov8的Detect层详解(输出维度改动)

我们可以在nn/modules.py中找到Detect()类，这里首先贴一下代码classDetect(nn.Module):"""YOLOv8Detectheadfordetectionmodels.
银嘉诚·2023-07-27 15:13

注意力机制（SE、Coordinate Attention、CBAM、ECA，SimAM）、即插即用的模块整理

总结曾经使用过的一些即插即用的模块以及一些注意力机制**注意力模块：SE**代码源自这位大佬的仓库：https://github.com/moskomule/senet.pytorchclassSELayer(nn.Module
吴大炮·2023-07-26 17:11

pytorch 中_call_impl()函数

记录pytorch版本中的nn.Module()重要函数1.
mingqian_chu·2023-07-26 04:55

深入浅出Pytorch函数——torch.nn.Module

实例importtorch.nnasnnimporttorch.nn.functionalasFclassModel(nn.Module):de
von Neumann·2023-07-25 01:41

pytorch工具——使用pytorch构建一个神经网络

目录构建模型模型中的可训练参数假设输入尺寸为32*32损失函数反向传播更新网络参数构建模型importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassNet(nn.Module
醋酸洋红就是我·2023-07-24 19:39

yolov5 加入可形变卷积

更新,之前的deformconv2d没加调制参数，应该是dcnv1现在的才是dcnv2fromtorchvision.opsimportDeformConv2dclassDCNConv(nn.Module
重糖爱好者·2023-07-23 03:46

4. BUILD THE NEURAL NETWORK

PyTorch中的每个模块都是nn.module的子类。神经网络是由其他模块（层）组成的模块本身。这种嵌套结构允许轻松地构建和管理复杂的体系结构。
绝园の暴风雨·2023-07-22 22:35

YOLOv8中的C2f的详细解读

首先是C2f的逻辑代码：classC2f(nn.Module):#CSPBottleneckwith2convolutionsdef__init__(self,c1,c2,n=1,shortcut=
AIOT魔法师·2023-07-21 11:08

YOLOv8的c2f模块详解

C2f模块的代码在YOLOv8上可以找到classC2f(nn.Module):"""CSPBottleneckwith2convolutions."""
下饭程序员王刚·2023-07-21 11:06

pytorch学习--第一个模型（线性模型）

能在迭代过程中使预测值和目标值能无限接近y=ωx+by=\omegax+by=ωx+b定义数据x=torch.rand([60,1])*10y=x*2+torch.randn(60,1)构建模型利用pytorch中的nn.Module
看星河的兔子·2023-07-20 22:53

ACSNet分割模型搭建

AdaptiveContextSelectionforPolypSegmentation源码：https://github.com/ReaFly/ACSNet.直接步入正题~~~一、基础模块classConvBlock(nn.Module
有温度的AI·2023-07-20 10:26

Swin Transformer

HierarchicalVisionTransformerusingShiftedWindows一、PatchEmbed（对应结构图中patchpartition和linearembedding）classPatchEmbed(nn.Module
有温度的AI·2023-07-20 10:56

MKDCNet分割模型搭建

://arxiv.org/abs/2206.06264v1源码：https://github.com/nikhilroxtomar/MKDCNet直接步入正题~~~一、基础模块classConv2D(nn.Module
有温度的AI·2023-07-20 10:56

LinkNet分割模型搭建

ExploitingEncoderRepresentationsforEfficientSemanticSegmentation直接步入正题~~~一、LinkNet1.decoder模块classDecoderBlock(nn.Module
有温度的AI·2023-07-20 10:25

TRT3-trt-basic - 5 插件的编写

生成onnx：还是一模一样的genonnx.pyclassModel(nn.Module):def__init__(self):super().
库达ZT·2023-07-20 03:24

随记·手撕coding | MultiheadAttention

importtorchimporttorch.nnasnnclassMultiheadAttention(nn.Module):#n_heads：多头注意力的数量#hid_dim：每个词输出的向量维度def
#苦行僧·2023-07-19 23:49

yolov5 自己训练的模型转换为onnx模型时，Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from 解决方案

Can'tgetattribute'SPPF'on解决办法如下：首先找到YOLOv5下的这个文件打开打开文件往下翻找到classSPP这一行，我的是在166行，在这一行上面添加下面的程序添加classSPPFclassSPPF(nn.Module
weixin_44482092·2023-07-19 18:54

【动手学深度学习】自定义层

动手学深度学习】自定义层不带参数的层 下面的类不带任何参数importtorchimporttorch.nn.functionalasFfromtorchimportnnclassCenteredLayer(nn.Module
MoonpieXia·2023-07-19 05:16

SRGAN记录

为图像对应的二进制位数，一般都是8（255位）2.SRGAN原理代码地址：https://github.com/aitorzip/PyTorch-SRGAN1）网络结构：classGenerator(nn.Module
Yankee_13·2023-07-19 04:52

使用torch.nn包来构建神经网络

一个nn.Module包含各个层和一个forward(input)方法，该方法返回output。
yanglamei1962·2023-07-18 16:25

Pytorch hook

BasicKnowledgeclassModule:classModel(nn.Module):Baseclassforallneuralnetworkmodules.Yourmodelsshouldalsosubclassthisclass.register_forward_hookisamethodinthisclassThehookwillbecalledeverytimeafter
Codiplay·2023-07-18 11:08

Pytorch如何打印与Keras的model.summary()类似的输出

pipinstalltorchsummary2.2代码importtorchimporttorch.nnasnnimporttorch.nn.functionalasFfromtorchsummaryimportsummaryclassNet(nn.Module
Xminyang·2023-07-18 06:08

CBAM非官方代码及官方代码

非官方代码两大模块:classChannelAttention(nn.Module):def__init__(self,in_planes,ratio=16):super(ChannelAttention
ENTRopYYY·2023-07-18 06:15

resnet 在残差结构内增加注意力机制并且预训练

所填添加的注意机制只是一个演示，怎么添加还需要自己琢磨classBasicBlock(nn.Module):expansion=1def__init__(self,inplanes,planes,stride
徒手写bug326·2023-07-18 06:15
上一页 3 4 5 6 7 8 9 10 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他