nn.Module

看代码学网络之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

cross-scale non-local attention

ImageSuper-ResolutionwithCross-ScaleNon-LocalAttentionandExhaustiveSelf-ExemplarsMining#cross-scalenon-localattentionclassCrossScaleAttention(nn.Module
SKY_yiyi_9·2020-09-16 22:20

Transformer的MultiHeadedAttention pytorch代码

classMultiHeadedAttention(nn.Module):"""Multi-HeadAttentionlayer:paramintn_head:thenumberofheads:paramintn_feat
·清尘··2020-09-16 19:49

focal loss

importtorchimporttorch.nnasnnclassFocalLoss(nn.Module):def__init__(self,gamma=0,eps=1e-7):super(FocalLoss
Bazinga521·2020-09-15 07:04

【PyTorch】模型构造：Module 和其子类Sequential、ModuleList、ModuleDict

只需重写forward方法classMLP(nn.Module):def__init__(self,**k
加油_加油_DrDu·2020-09-15 01:54

使用 pytorch 保存和加载训练好的模型

定义网络结构这里使用最简单的一个结构，两个线性的全连接层，有激活函数，用来拟合二维空间上的一些点"""net.py用于定义网络的结构"""classNet(nn.Module):def__init__(
假装很坏的谦谦君·2020-09-15 00:16

nn.Module -使用Module 类来自定义网络

1前言：如何自定义一个模型-通过继承nn.Module类来实现，在__init__构造函数中申明各个层的定义，在forward中实现层直接的连接关系，实际上就是前向传播的过程。
368chen·2020-09-14 19:04

PyTorch复现经典网络————AlexNet

importtorchfromtorchimportnnimportnumpyasnpfromtorch.autogradimportVariablefromtorchvision.datasetsimportCIFAR10classAlexNet(nn.Module
起步晚就要快点跑·2020-09-14 18:35

mini-batch的TripleLoss实现(Pytorch)

classTripleLoss(nn.Module):def__init__(self,margin=0.3):super(TripleLoss,self).__init__()s
dc90000·2020-09-14 17:28

pytorch入坑三 | nn.module

文章目录1.两个基本结构1.1Parameter(参数)1.2Containers(容器)1.2.1Module(模型）核心功能查看模块类型转换设置功能注册功能1.2.2Sequential(序列)1.2.3ModuleList模型列表1.2.4ParameterList参数列表2.常用的网络层之前我们介autograd,但对于比较大的复杂网络来说，autograd会显得有点太底层了，而在大多数的
我是天才很好·2020-09-14 16:23
上一页 23 24 25 26 27 28 29 30 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他