pytorch实现特殊的Module--Sqeuential三种写法
2020-02-13 10:29
751 查看
我就废话不多说了,直接上代码吧!
# -*- coding: utf-8 -*- #@Time :2019/7/1 13:34 #@Author :XiaoMa import torch as t from torch import nn #Sequential的三种写法 net1=nn.Sequential() net1.add_module('conv',nn.Conv2d(3,3,3)) #Conv2D(输入通道数,输出通道数,卷积核大小) net1.add_module('batchnorm',nn.BatchNorm2d(3)) #BatchNorm2d(特征数) net1.add_module('activation_layer',nn.ReLU()) net2=nn.Sequential(nn.Conv2d(3,3,3), nn.BatchNorm2d(3), nn.ReLU() ) from collections import OrderedDict net3=nn.Sequential(OrderedDict([ ('conv1',nn.Conv2d(3,3,3)), ('bh1',nn.BatchNorm2d(3)), ('al',nn.ReLU()) ])) print('net1',net1) print('net2',net2) print('net3',net3) #可根据名字或序号取出子module print(net1.conv,net2[0],net3.conv1)
输出结果:
net1 Sequential( (conv): Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1)) (batchnorm): BatchNorm2d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (activation_layer): ReLU() ) net2 Sequential( (0): Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1)) (1): BatchNorm2d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (2): ReLU() ) net3 Sequential( (conv1): Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1)) (bh1): BatchNorm2d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (al): ReLU() ) Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1)) Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1)) Conv2d(3, 3, kernel_size=(3, 3), stride=(1, 1))
以上这篇pytorch实现特殊的Module--Sqeuential三种写法就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- JavaScript模拟实现封装的三种方式及写法区别
- 单例模式(Singleton pattern)的三种写法(含枚举实现方式)
- 二叉树三种遍历方式的递归和非递归写法(python 实现)
- Dagger2 同时引用多个 Module 的三种写法
- SQL调优日记之发挥SQL性能与你的写法有关--对比三种方式实现相同功能
- CSS特殊样式(二)三种css方法实现三角效果
- 三种方式实现数组去重
- 跨域的三种实现方式
- 使用PHP实现水仙花数及各种特殊有趣数的输出
- net中 类、方法作为参数传递的写法整理。(通过反射实现)(即传递类、传递方法的实现)
- 三种java实现多线程的方式
- Android_自定义View的三种实现方式
- 三种java编程方法实现斐波那契数列
- Java利用 DES / 3DES / AES 这三种算法分别实现 对称加密
- 大文件上传的HttpModule实现。(一)
- 单点登录的三种实现方式
- Poj1151/HDU1542(求并矩形覆盖的面积)分别用离散化处理,矩形切割,线段树三种方法实现.
- 资源 | Github项目:斯坦福大学CS-224n课程中深度NLP模型的PyTorch实现
- Linux中实现GetModuleFileName
- JQuery实现表格的相同单元格合并的三种方法