nn.Module

PyTorch dropout设置训练和测试模式的实现

看代码吧~classNet(nn.Module):…model=Net()…model.train()#把module设成训练模式，对Dropout和BatchNorm有影响model.eval()#把
·2021-05-27 13:10

pytorch中LN(LayerNorm)及Relu和其变相的输出操作

importtorchimporttorch.nnasnnimporttorch.nn.functionalasFclassmodel(nn.Module):def__init__(self):super
·2021-05-27 11:23

pytorch 实现冻结部分参数训练另一部分

比如加载了resnet预训练模型之后，在resenet的基础上连接了新的模快，resenet模块那部分可以先暂时冻结不更新，只更新其他部分的参数，那么可以在下面加入上面那句话classRESNET_MF(nn.Module
·2021-05-23 18:25

pytorch中的model.eval()和BN层的使用

看代码吧~classConvNet(nn.module):def__init__(self,num_class=10):super(ConvNet,self).
·2021-05-22 13:11

pytorch 中nn.Dropout的使用说明

看代码吧~ClassUSeDropout(nn.Module):def__init__(self):super(DropoutFC,self).
·2021-05-20 12:20

【pytorch】模型的搭建保存加载

classLeNet(nn.Module):def__init__(self):super(LeNet,self).
DeepWeaver·2021-05-16 14:43

openCV调用YOLOv5 ONNX模型

YOLOv5的ONNX模型，openCV直接读取会失败，原因在于Pytorch2ONNX不支持对slice对象赋值，需要更改下列两个地方代码：yolov5/models/common.pyclassFocus(nn.Module
ghang35·2021-05-10 12:07

BAM注意力机制模块

参考链接:BAMandCBAM.实验代码:importtorchimportmathimporttorch.nnasnnimporttorch.nn.functionalasFclassFlatten(nn.Module
敲代码的小风·2021-05-07 16:02

浅谈pytorch中的nn.Sequential(*net[3: 5])是啥意思

看到代码里面有这个1classResNeXt101(nn.Module):2def__init__(self):3super(ResNeXt101,self).
·2021-04-26 10:56

看代码学网络之GoogLeNet

model.pyimporttorch.nnasnnimporttorchimporttorch.nn.functionalasFclassGoogLeNet(nn.Module):def__init_
monthmonthup·2021-04-22 21:51

看代码学网络之VGGNet

model.pyimporttorch.nnasnnimporttorchclassVGG(nn.Module):def__init__(self,features,num_classes=1000,init_weights
monthmonthup·2021-04-22 15:28

PyTorch手动初始化卷积层并进行卷积运算报数据类型错误Expected object of scalar type Double but got scalar type Float for arg

ExpectedobjectofscalartypeDoublebutgotscalartypeFloatforargument#4‘bias’尝试脱离nn.Module模块手动定义conv2d层，并对权重
黑夜里游荡·2021-04-22 14:37

简单聊聊nn.ModuleList

和普通list不一样，它和torch的其他机制结合紧密，继承了nn.Module的网络模型class可以使用nn.ModuleList并识别其中的parameters普通list中的子module并不能被主
吉米_王·2021-04-18 14:18

PyTorch建立RNN相关模型

建立RNN模型#定义网络结构classNet(nn.Module):def__init__(self):super(Net,self).
·2021-04-06 13:24

PyTorch建立RNN相关模型

建立RNN模型#定义网络结构classNet(nn.Module):def__init__(self):super(Net,self).
·2021-04-06 13:03

看代码学网络之AlexNet

model.pyimporttorch.nnasnnimporttorchclassAlexNet(nn.Module):def__init__(self,num_classes=1000,init_weights
monthmonthup·2021-03-31 09:20

PYTORCH BN层

1、代码classNet(nn.Module):def__init__(self):#对所有的层初始化super(Net,self).__init__()#父类的所有的属性self.conv1
@慕云·2021-03-18 21:16

pytorch dropout代码解读

1、代码classNet(nn.Module):def__init__(self):#对所有的层初始化super(Net,self).__init__()#父类的所有的属性self.conv1
远方以南·2021-03-18 18:47

self-attention pytorch实现

AttentionclassAttention(nn.Module):def__init__(self,dim,num_heads=2,qkv_bias=False,qk_scale=None,attn_drop
lowl·2021-03-18 17:21

PyTorch中nn.Module理解

nn.Module是Pytorch封装的一个类，是搭建神经网络时需要继承的父类：importtorchimporttorch.nnasnn#括号中加入nn.Module（父类）。
菜鸟向前冲fighting·2021-03-17 21:13

关于Transformer训练的问题

TheTheAnnotatedTransformer》教程写下来，卡在了GreedyDecoding部分，回溯发现问题出在两个部分在Embeddings部分，教程中如下classEmbeddings(nn.Module
我的昵称违规了·2021-03-11 19:34

BackBone

BackBoneBackBoneBackBoneimporttorchimporttorch.nnasnnimporttorch.nn.functionalasFimportmathfromcollectionsimportOrderedDict1.MISH激活函数classMish(nn.Module
茗君（Major_s）·2021-03-07 14:50

pytorch 中forward 的用法与解释说明

前言最近在使用pytorch的时候，模型训练时，不需要使用forward，只要在实例化一个对象中传入对应的参数就可以自动调用forward函数即：forward的使用classModule(nn.Module
·2021-02-26 12:51

python正弦函数拟合_使用 pytorch 创建神经网络拟合sin函数的实现

1)构建数据集，包括输入，对应的标签y2)构建神经网络模型，一般基于nn.Module继承一个net类，必须的是__init__函数和
牛耿·2021-02-21 04:54

最简单的神经网络

noise=torch.randn(50)y=2*x+3+noise.numpy()绘制x，y的图象如下构建神经网络torch里面可以基于nn.Module类写自己的神经网络，这里使用最简单的
rivrui·2021-02-04 21:41

nn.Module源码介绍(三)---模型加载/保存乱炖

前言断断续续写了好久，本篇是最后一篇。太肝了，写这东西,写这东西收视率不高，实在不行就换实战把。文章目录前言一、创建一个简单网络并训练保存二、State_dict()源码解读三、加载模型四、实战：保存/加载（模型+优化器+epoch）五、网络相同但命名方式不同的加载解析六、网络不同的权重加载解析附录、其他常用的代码片段参考文献一、创建一个简单网络并训练保存importtorchimporttorc
武乐乐~·2021-01-29 16:31

Faster R-CNN ROIAlign TwoMLPHead FastRCNNPredictor

bilibilifaster_rcnn_framework.pyclassFasterRCNNBase(nn.Module):"""MainclassforGeneralizedR-CNN.Arguments
蓝鲸鱼BlueWhale·2021-01-29 14:16

Faster R-CNN RPN

bilibilirpn_function.py一、RPNheadclassRPNHead(nn.Module):"""addaRPNheadwithclassificationandregression
蓝鲸鱼BlueWhale·2021-01-26 21:43

Faster R-CNN GeneralizedRCNNTransfrom

bilibilitransform.pyclassGeneralizedRCNNTransform(nn.Module):"""Performsinput/targettransformationbeforefeedingthedatatoaGeneralizedRCNNmodel.Thetransformationsitperformare
蓝鲸鱼BlueWhale·2021-01-26 20:23

Faster R-CNN框架

bilibilifaster_rcnn_framework.pyclassFasterRCNNBase(nn.Module):"""MainclassforGeneralizedR-CNN.Arguments
蓝鲸鱼BlueWhale·2021-01-26 20:31

学习笔记2：Pytorch 神经网络初始化权重与偏置

classCONCAT_CNN(nn.Module):def__init__(self):super(CONCAT_CNN,self).
只想飘一飘·2021-01-20 15:48

nn.Module源码介绍(二)--冻结参数篇

Module源码介绍本篇是nn.Module源码系列文章介绍第二篇，建议读者在阅读第一篇之后，在来阅读本篇。当然，也可以直接阅读本文，因为提供了大量的实例。
武乐乐~·2021-01-17 20:42

FM代码

importtorch.nnasnnfromscipy.sparseimportcsrimportpandasaspdimportnumpyasnpimporttorchfromtqdmimporttqdm#FMmodelclassFM_model(nn.Module
Take your time_·2021-01-17 19:23

Pytorch什么时候开始调用forward

importtorchfromtorchimportnnclassMLP(nn.Module):#声明带有模型参数的层，这里声明了两个全连接层def__init__(self,**kwargs):#调用
gz153016·2021-01-16 23:51

nn.Module介绍(一)

目录文章目录目录前言一、构造一个极简网络二、初始化部分介绍三、添加一个卷积层的简单网络四、通过简单网络洞察nn.Module的其他属性和方法总结前言一、构造一个极简网络importtorchimporttorch.nnasnnfromcollections.abcimportIterable
武乐乐~·2021-01-11 21:04

Pytorch模型加载保存

首先定义一个名为Example的模型importtorch.nnasnnimporttorch.nn.functionalasFimporttorch.optimasoptimclassExample(nn.Module
jhsignal·2020-12-31 10:24

FPN的代码理解（可视化）

Inc.anditsaffiliates.AllRightsReserved.importtorchimporttorch.nn.functionalasFfromtorchimportnnclassFPN(nn.Module
G果·2020-12-29 12:41

tensorboard 数据降维分布图_PyTorch TensorBoard 可视化数据和模型训练过程

在60分钟闪电战中，我们展示了如何加载数据，如何为我们定义为nn.Module的子类的模型提供数据，如何在训练数据上训练此模型，并在测试数据上测试它。
weixin_39553757·2020-12-19 23:23

PyTorch可视化模型结构

torchviz代码：importtorchimportnumpyasnpimporttorch.nnasnnimporttorch.nn.functionalasFimporttorchvizclassNet(nn.Module
烟雨风渡·2020-12-18 09:56

pytorch 限制gpu的使用量_PyTorch踩过的坑

对于nn.Module:mod
weixin_39593354·2020-11-27 23:31

python出现no module named cv2_pytorch源码阅读（一）Module类

本文针对pytorch的核心基类nn.Module【1】进行分析，Module作为各种操作的父类是每个网络定义必须继承的基类。
weixin_39574943·2020-11-26 16:07

torch.nn.Module模块简单介绍

torch.nn是专门为神经网络设计的模块化接口，nn.Module是nn中十分重要的类。
allan2222·2020-11-23 16:29

（pytorch-深度学习）SE-ResNet的pytorch实现

SE-ResNet的pytorch实现残差块：classResiual_block(nn.Module):def__init__(self,in,middle_out,out,kernel_size=3
我是一颗棒棒糖·2020-11-20 10:39

pytorch使用tensorboardX做可视化（一）loss和直方图

tensorboardX相关的语句都标记了出来，主要是传个每轮的loss，参数model自带importtorch.nnasnnimportnumpyasnpimporttorch#定义lenet5classLeNet5(nn.Module
佳hong·2020-11-07 20:33

【ai竞赛系列】新冠疫情预测--时序数据预测模型

官方给的历史数据如下：后面采用的是LSTM进行预测，PyTorch实现：classLSTMpred(nn.Module):def__
小可爱本人^_^·2020-11-05 12:35

self-attention实现

ChannelDomainsoft-attention是可微的，可以通过梯度来实现importtorchimporttorch.nnasnnimportnumpyasnpimportmathclassSelfAttention(nn.Module
清晨我上马·2020-11-04 15:20

5.2 使用pytorch搭建GoogLeNet网络 笔记

B站资源csdn本家文章目录modeltrainpredictmodel1.BasicConv2d类2.Inception3.InceptionAux(nn.Module):#辅助分类器4.GoogLeNet
新川宝宝·2020-10-22 11:57

Pytorch框架学习 -2 torch.nn.modules.Module(nn.Module)理解

文章目录Pytorch框架学习-2torch.nn.modules.Module(nn.Module)理解最简单的例子分析部分源码：基本参数dump_patches_versiontraining初始化函数
goes_on·2020-10-15 15:40

【经典网络结构实现】LeNet-5

5网络的简单实现LeNet-5应用于手写数字识别Pytorch实现手写数字识别LeNet-5网络的简单实现#导入相应的包importtorch.nnasnn#搭建LeNet-5classLeNet5(nn.Module
MING.MING·2020-10-10 11:55

PyTorch实现ResNet18

ResNet-18结构基本结点代码实现importtorchimporttorch.nnasnnfromtorch.nnimportfunctionalasFclassRestNetBasicBlock(nn.Module
一个不想写代码的程序员·2020-09-30 09:33
上一页 25 26 27 28 29 30 31 32 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他