nn.Module

ResNet34 构建（继续优化代码。。）

importtorchfromtorchimportnnclassidendity_mapping_block(nn.Module):def__init__(self,input_channels,output_channels
Jay_Mapp·2023-01-07 20:44

Pytorch:函数的手动梯度计算方法

学习torch,掌握tensor,autograd,Variable,nn,nn.Module,loss,optim,Dat
三世·2023-01-07 20:42

PyTorch定义模型结构的常用方法

Module类的使用方法Module类的add_module()方法importtorch.nnasnn#继承nn.Module类，构建网络模型classLogicNet(nn.Module):def_
hithithithithit·2023-01-07 17:38

Pytorch 神经网络实践

importtorchimporttorch.nnasnnimporttorch.nn.functionalasFfromtorchsummaryimportsummary#定义网络类classNet1(nn.Module
StartTodayComeOn·2023-01-07 15:20

PyTorch使用技巧2：netron可视化模型结构

importtorchimporttorch.nnasnnimporttorch.nn.functionalasFimporttorch.onnximportnetronclassmodel(nn.Module
源代码杀手·2023-01-07 14:43

Paddle：reshape算子与pytorch中的view算子

reshape算子#pytorch#AvgPool2d比AdaptiveAvgPool2d更快,但是使用View和Mean会比AvgPool2d快5倍.classFastGlobalAvgPool2d(nn.Module
微风❤水墨·2023-01-07 07:04

SPP的理解

值得注意的是：虽然使用了不同大小的池化核，但池化前后每条分支的高宽一致，因为stride=1，padding=(kernel_size-1)/2=kerner//2classSPPBottleneck(nn.Module
沐辰光·2023-01-06 22:35

CycleGAN模型——pytorch实现

#输入图像shape默认为(3,256,256)classDiscriminator(nn.Module):#定义判别器def__init__(self):#初始化方法super(Discriminator
Peach_____·2023-01-06 22:48

[The Annotated Transformer] 代码修正

1.RuntimeError:"exp"notimplementedfor'torch.LongTensor'classPositionalEncoding(nn.Module)div_term=torch.exp
weixin_30278237·2023-01-06 20:59

HorNet+YOLOv5改进方案

理论请参考论文：https://arxiv.org/pdf/2207.14284.pdfyolov5加HorNet模块代码1.在common.py文件中添加如下代码部分代码如下classGnconv(nn.Module
Dropout~~·2023-01-06 14:43

ConvNeXt+YOLOv7改进方案

理论请参考论文：[2201.03545]AConvNetforthe2020s(arxiv.org)yolov7添加CNeB模块代码1.在common.py文件中添加如下代码classLayerNorm_s(nn.Module
Dropout~~·2023-01-06 14:43

CCNet YOLOv5目标检测实现

目前还没有进行训练，yolo验证无误基于此文章进行修改CCNetclassCCBottleneck(nn.Module):def__init__(self,c1,c2,shortcut=True,g=1
hgzyh123·2023-01-06 14:43

Pytorch中查看模型参数信息, net.parameters()、net.named_parameters(), net.layer.parameters()等,末尾有，Variable，hook

测试代码由此开始：classNet(nn.Module):def__init__(self):super(Net,self).
生气的小木块·2023-01-06 11:05

（机器学习深度学习常用库、框架|Pytorch篇）第（待定）节：卷积神经网络CNN中一些经典网络结构写法

下图显示了其结构：输入的二维图像，先经过两次卷积层到池化层，再经过全连接层，最后使用softmax分类作为输出层网络结构classLeNet5(nn.Module):"""使用sequential构建
快乐江湖·2023-01-06 11:59

PyTorch torch.nn从头开始【详解附代码】

关于torch.nn文章目录关于`torch.nn`MNIST数据下载1.Neuralnetfromscratch(不用torch.nn)2.Usingtorch.nn.functional3.使用nn.Module
qq_39809262·2023-01-06 11:15

详解Pytorch中的网络构造

背景在PyTroch框架中，如果要自定义一个Net(网络，或者model，在本文中，model和Net拥有同样的意思)，通常需要继承自nn.Module然后实现自己的layer。
anshiquanshu·2023-01-06 09:35

pytorch_神经网络模型搭建系列（2）：自定义神经网络层

目录1、nn.Module——搭建属于自己的神经网络1.1回顾系统预定义的层1.1.1最常用的Linear层1.1.2Conv2d类的源代码1.1.3小结1.1.4自定义层的基本步骤1.2简单实现-自定义层
xgxg1314·2023-01-05 23:28

Pytorch学习日记——nn.Module的使用

学习视频——B站【小土堆】代码importtorchfromtorchimportnnclassTudui(nn.Module):def__init__(self):super().
快乐白给·2023-01-05 19:42

一步步搭建可识别CIFAR10中10个类别的卷积神经网络（基于Pytorch）|深度学习入门项目

classMayu(nn.Module):def__init__(self):super(Mayu,self).__init__(
Mr.Mayuyuyuyu·2023-01-05 14:18

【torch】自定义卷积权重

自定义卷积权重importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassCNN(nn.Module):def__init__(self,
椰子奶糖·2023-01-05 03:19

[2021ICCV]Swin Transformer模型的一些模块

GitHubCSDNPatchEmbed:将输入的图片进行切分classPatchEmbed(nn.Module):"""2DImagetoPatchEmbedding"""def__init__(self
清欢年岁~·2023-01-04 20:01

使用GPU训练模型

一般通过nn.Module来构建模型并编写自定义训练循环。为了更加方便地训练模型，作者编写了仿keras的Pytorch模型接口：torchkeras，作为Pytorch的高阶API。
Python_Ai_Road·2023-01-04 18:10

VAE(Variational Autoencoder)简单记录

理论代码以下代码来源:https://zhuanlan.zhihu.com/p/151587288classVAE(nn.Module):def__in
Andy Dennis·2023-01-04 16:12

GAN模型——pytorch实现

classDiscriminator(nn.Module):#定义判别器def__init__(self,img_size=(28,28)):#初始化方法super(Discriminator,self
Peach_____·2023-01-04 15:31

CGAN模型——pytorch实现

classDiscriminator(nn.Module):#定义判别器def__init__(self,img_size=(28,28),num_classes=2):#初始化方法super(Discriminator
Peach_____·2023-01-04 15:31

ACGAN模型——pytorch实现

classDiscriminator(nn.Module):#定义判别器def__init__(self,img_size=(64,64),num_classes=2):#初始化方法super(Discriminator
Peach_____·2023-01-04 15:31

DCGAN模型——pytorch实现

classDiscriminator(nn.Module):#定义判别器def__init__(self,img_size=(64,64)):#初始化方法super(Discriminator,self
Peach_____·2023-01-04 15:28

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

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

Alexnet详解

model.pyimporttorch.nnasnnimporttorchclassAlexNet(nn.Module):def__init__(self,num_classes=1000,init_weights
Recursi·2023-01-04 09:52

PyTorch神经网络+VGG

PyTorch神经网络:importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassNet(nn.Module):def__init__(
weixin_58565789·2023-01-03 22:51

pytorch 将模型作为特征提取器（提取中间层特征）

classMyModel(nn.Module):def__init__(self):super(MyModel,self).
无名草鸟·2023-01-03 22:19

【目标检测】|yolov6 结构代码分析

yolov6s整体：model/yolo.pyModel:classModel(nn.Module):def__init__():self.backbone,self.neck,self.detect=
rrr2·2023-01-03 09:33

【技术文档】HRNet姿态估计

目录模型结构augmentcfgloss模型结构classPoseHighResolutionNet(nn.Module):def__init__(self,cfg,**kwargs):self.inplanes
superjfhc·2023-01-03 07:53

【神经网络常用损失函数】

神经网络常用损失函数FocalLossclassFocalLoss(nn.Module):#Wrapsfocallossaroundexistingloss_fcn(),i.e.criteria=FocalLoss
小小小MaYi·2023-01-03 01:31

语义分割任务中如何处理label为255的标签

训练计算loss时的处理importtorchfromtorchimportnnclassCrossEntropy2d(nn.Module):def__init__(self,ignore_label=
算法菜鸟飞高高·2023-01-02 14:03

ResNet模型——pytorch实现

classBasicBlock(nn.Module):#定义残差块，resnet18、resnet34使用此残差块expansion=1#残差操作维度变化倍数def__init__(self,in_channel
Peach_____·2023-01-02 10:42

VGG模型——pytorch实现

classVGG(nn.Module):#定义VGG模型类def__init__(self,features,num_classes=2,init_weights=False):#定义初始化方法super
Peach_____·2023-01-02 10:40

【Pytorch学习】add_module()函数用法

add_module就好像是list的用法，看代码之后就懂了（代码来自文献【3】）fromtorchimportnnfromtorchsummaryimportsummaryclassNet_test(nn.Module
一穷二白到年薪百万·2023-01-02 08:32

基于PyTorch实现 Inception-ResNet-v1

Inception-ResNetandtheImpactofResidualConnectionsonLearning2、网络框架本文github链接Stem1.框架图2.代码实现因为要经常用，所以先定义一个3x3卷积，1x1卷积classconv3x3(nn.Module
NAND_LU·2023-01-01 21:08

pytorch搭建YOLOv5网络

1.基本组成模块1.1Foucs模块#--------------------------Foucs模块-------------——-classFoucs(nn.Module):def__init__
weixin_45639189·2023-01-01 14:01

图像注意力机制

importtorchimporttorch.nnasnnimportmathclassse_block(nn.Module):def__init__(self,channel,ratio=16):su
chairon·2023-01-01 14:51

pytorch保存参数及模型的两种方式

classLinear(nn.Module):def__init__(self):super().
Bi 8 Bo·2023-01-01 10:49

pytorch: nn.Dropout 和 nn.functional.dropout 的区别

pytorch:nn.Dropout和nn.functional.dropout的区别importtorchimporttorch.nnasnnclassFunctionalDropout(nn.Module
一只楚楚猫·2023-01-01 09:58

深度学习：04 神经网络的搭建

nn.Module是PyTorch
夏天是冰红茶·2023-01-01 07:04

model.parameters()的理解与使用

首先定义一个模型#designModelclassNeuralNetwork(nn.Module):def__init__(self):super(NeuralNetwork,self).
Httoti·2022-12-31 15:24

pytorch应用（入门2） 一维线性回归

目录nn.Module（模组）torch.optim(优化)模型的保存和加载一维线性回归代码如下：均方差损失函数nn.MSELoss()model.parameters()的理解与使用torch.autograd.Variableoptimizer.zero_grad
懒回顾，半缘君·2022-12-31 15:44

transformer中

中前馈全连接层就是具有两层线性层的全连接网络前馈全连接层的作用考虑注意力机制可能对复杂过程的拟合程度不够，通过增加两层网络来增强模拟的能力#前馈全连接层的代码分析#通过类PositionwiseFeedForward来实现前馈全连接层classPositionwiseFeedForward(nn.Module
weixin_44522477·2022-12-31 12:36

PyTorch 3 - 模型相关方法

多分类模块的组合ONNX将pytorchmodel转为onnx格式将caffee2导入onnx模型使用onnx将pytorch模型迁移到CoreML创建模型如，创建线性模型classLinearModel(nn.Module
伊织code·2022-12-30 14:13

图神经网络16-DGL实战：构建图神经网络（GNN）模块

importtorch.nnasnnfromdgl.utilsimportexpand_as_pairclassSAGEConv(nn.Module):def__init__(self,in_feats
致Great·2022-12-30 08:38

32李沐动手学深度学习v2/含并行连结的网络，GoogLeNet

importtorchfromtorchimportnnfromtorch.nnimportfunctionalasFfromd2limporttorchasd2l#GoogLeNet的v1版本的inception块classInception(nn.Module
xcrj·2022-12-29 23:08
上一页 9 10 11 12 13 14 15 16 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他