pytorch实现特殊的Module--Sqeuential三种写法


Posted in Python onJanuary 15, 2020

我就废话不多说了,直接上代码吧!

# -*- 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三种写法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python修改Excel数据的实例代码
Nov 01 Python
python中的__init__ 、__new__、__call__小结
Apr 25 Python
简单介绍Python中的filter和lambda函数的使用
Apr 07 Python
简单介绍Python中的readline()方法的使用
May 24 Python
Python基于更相减损术实现求解最大公约数的方法
Apr 04 Python
基于多进程中APScheduler重复运行的解决方法
Jul 22 Python
python3使用print打印带颜色的字符串代码实例
Aug 22 Python
python 表格打印代码实例解析
Oct 12 Python
python argparser的具体使用
Nov 10 Python
python3中rank函数的用法
Nov 27 Python
Python json读写方式和字典相互转化
Apr 18 Python
使用pycharm和pylint检查python代码规范操作
Jun 09 Python
python实现删除列表中某个元素的3种方法
Jan 15 #Python
python opencv根据颜色进行目标检测的方法示例
Jan 15 #Python
Python基于Tensor FLow的图像处理操作详解
Jan 15 #Python
OpenCV哈里斯(Harris)角点检测的实现
Jan 15 #Python
Pytorch模型转onnx模型实例
Jan 15 #Python
Python通过TensorFLow进行线性模型训练原理与实现方法详解
Jan 15 #Python
详解Python实现进度条的4种方式
Jan 15 #Python
You might like
重料打造自己的“宝马”---第三代
2021/03/02 无线电
用php将任何格式视频转为flv的代码
2009/09/03 PHP
php无序树实现方法
2015/07/28 PHP
PHP+iframe图片上传实现即时刷新效果
2016/11/18 PHP
Laravel源码解析之路由的使用和示例详解
2018/09/27 PHP
PHP反射实际应用示例
2019/04/03 PHP
jQuery EasyUI API 中文文档 - Spinner微调器使用
2011/10/21 Javascript
javascript 弹出的窗口返回值给父窗口具体实现
2013/11/23 Javascript
从数据库读取数据后将其输出成html标签的三种方法
2014/10/13 Javascript
JS实现黑色大气的二级导航菜单效果
2015/09/18 Javascript
Javascript自执行匿名函数(function() { })()的原理浅析
2016/05/15 Javascript
省市区三级联动jquery实现代码
2020/04/15 Javascript
Javascript之面向对象--接口
2016/12/02 Javascript
Bootstrap弹出框(Popover)被挤压的问题小结
2017/07/11 Javascript
使用nodeJs来安装less及编译less文件为css文件的方法
2017/11/20 NodeJs
vue表单自定义校验规则介绍
2018/08/28 Javascript
微信小程序npm引入vant-weapp的踩坑记录
2019/08/01 Javascript
中级前端工程师必须要掌握的27个JavaScript 技巧(干货总结)
2019/09/23 Javascript
javascript 模块依赖管理的本质深入详解
2020/04/30 Javascript
针对Vue路由history模式下Nginx后台配置操作
2020/10/22 Javascript
浅析Python中signal包的使用
2015/11/13 Python
python的Tqdm模块的使用
2018/01/10 Python
python实现堆和索引堆的代码示例
2018/03/19 Python
python3 pandas 读取MySQL数据和插入的实例
2018/04/20 Python
python ddt数据驱动最简实例代码
2019/02/22 Python
使用matplotlib绘制图例标签中带有公式的图
2019/12/13 Python
python 伯努利分布详解
2020/02/25 Python
如何基于python3和Vue实现AES数据加密
2020/03/27 Python
浅析Python __name__ 是什么
2020/07/07 Python
Python 中 sorted 如何自定义比较逻辑
2021/02/02 Python
一款利用html5和css3动画排列人物头像的实例演示
2014/12/05 HTML / CSS
LTD Commodities:礼品,独特发现,家居装饰,家用器皿
2017/08/11 全球购物
Shopee新加坡:东南亚与台湾电商平台
2019/01/25 全球购物
创立科技Java面试题
2015/11/29 面试题
融资租赁计划书
2014/04/29 职场文书
人口与计划生育责任书
2015/05/09 职场文书