nn.Module

[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

PyTorch-07 卷积神经网络（什么是卷积、卷积神经网络、池化层、Batch normalization、经典卷积网络、深度残差网络 ResNet、nn.Module、数据增强）

PyTorch-07卷积神经网络（什么是卷积、卷积神经网络、池化层、Batchnormalization、经典卷积网络(LeNet-5、AlexNet、VGG、GoogLeNet)、深度残差网络ResNet、nn.Module
Henrik698·2022-12-29 12:04

Pix2Pix原理解析以及代码流程

文章目录1、网络搭建2、反向传播过程3、PatchGAN4.与CGAN的不同之处1、网络搭建classUnetGenerator(nn.Module):"""CreateaUnet-basedgenerator
啊菜来了·2022-12-29 12:33

猫狗大战(kaggle竞赛-猫狗图像分类)

中猫狗大战中的部分数据集（2000张训练数据+500张测试数据）本次实验中使用了DNN、CNN、RNN分别进行了图像识别，具体代码如下：DNN模型：全连接层神经元个数FC1512FC2128FC32classDNN(nn.Module
lingffffighting·2022-12-29 04:06

基于中心对齐的领域泛化损失

classCenter_loss(nn.Module):def__init__(self,src_class):super(Center_loss,self).
好人就是拉风·2022-12-29 00:21

Metric learning

一：减小类内距离，增大类间距离classMetric_loss(nn.Module):def__init__(self,src_class):super(Metric_loss,self).
好人就是拉风·2022-12-29 00:51

pytorch 关于权重（初始化、固定）

权重初始化：emb那种可以直接from_pretrained固定权重利用requires_grad=False，这样只是不更新，但是还是会计算并占用显存classNet(nn.Module):def__
爱coding的白兰客·2022-12-28 21:52

Pytorch如何约束神经网络中权重/偏执的范围

笔者由于在做可解释机器学习，后来在pytorch的discussion上终于找到了相应的解决方案，这里给大家看下方法一：首先编写模型结构：classModel(nn.Module):def__init_
Geeksongs·2022-12-28 21:15

关于torch.round可导性问题

先上结论：不可导importtorchimporttorchimporttorch.nnasnnclassg(nn.Module):def__init__(self):super(g,self).
撒旦即可·2022-12-28 16:01

pytorch模型构建、训练、测试及预测

一神经网络的典型处理流程1.定义可学习参数的网络结构（堆叠各层和层的设计）；继承nn.Module模块，改写forward方法。
myyzqt·2022-12-28 13:52

Pytorch初始化模型权重最快捷的方法

Example:创建一个网络，其中包括一个预先定义的DoubleConv类classDoubleConv(nn.Module):def__init__(self,in_channels,out_channels
Rainlin.Zhang·2022-12-28 10:16

PyTorch学习笔记：针对一个网络的权重初始化方法

#针对一个网络的权重初始化方法importtorchimporttorch.nnasnn##建立一个测试网络classTestNet(nn.Module):def__init__(self):super
code_carrot·2022-12-28 10:44

pytorch-网络层初始化记录

初始化：pytorch官网有关自然语言处理-权重初始化definit_weights(m:nn.Module):forname,paraminm.named_parameters():if'weight'inname
北交_段宇航·2022-12-28 10:14

PyTorch 的容器 Container 包括 ( nn.ModuleList, nn.ModuleDict, nn.Sequential )

PyTorch几种构建网络的容器类classMyModel1(nn.Module):def__init__(self):super(MyModel,self).
峡谷的小鱼·2022-12-28 07:44

9 - nn.Module、nn.Sequential以及其他容器Module的源码讲解

文章目录1.train训练模式2.eval推理模式3.requires_grad_4.zero_grad5.str(mymodel)6.nn.Sequential7.nn.ModuleList8.nn.ModuleDict9.nn.ParameterList10.nn.ParameterDict1.train训练模式将模型设置成训练模式，本质上式将self.training=True启用dropo
取个名字真难呐·2022-12-28 07:40

Pytorch nn.Module源码解析

官网介绍：https://pytorch.org/docs/stable/generated/torch.nn.Module.html；nn.Module模块是所有神经网络的基类，任何类都应该继承这个nn.Module
不存在的c·2022-12-28 07:00

Pytorch容器：nn.Sequential，ModuleList，ParameterList源码解析

__file__#container.py包含内容#都继承自nn.Module类#classSequential(Module)#classModuleList(Module)#classModuleDict
不存在的c·2022-12-28 07:28

Pytorch英文官方文档学习笔记(三、Torch.nn及torch.optim)

一、nn.Module的使用EverymoduleinPyTorchsubclassesthenn.Module自己定义的每个module都一定是nn.Module的子类pytorch在nn.Module
InsaneGe·2022-12-28 01:25

torch自带插值函数

classModel(nn.Module):#2022.11.7修改前，这个Model能跑通#forMultivariatedef__init__(self,configs,channel=96,ratio
思考实践·2022-12-28 01:20

basic_vsr 代码介绍

.basic_vsr推理结果：本文参考basic_vsrcode:githubcode1.basic_vsr使用的一些模块介绍：#1）一个光流预测的网络，预测相邻两帧的flowclassSPyNet(nn.Module
tony365·2022-12-27 21:07

Pytorch实现ChannelShuffle(通道混洗)

importtorchimporttorch.nnasnnclassChannel_Shuffle(nn.Module):def__init__(self,num_groups):super(Channel_Shuffle
FoerKent·2022-12-27 14:12
上一页 9 10 11 12 13 14 15 16 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他