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实现数通设备tftp备份配置文件示例
Apr 02 Python
Python入门篇之正则表达式
Oct 20 Python
Python动刷新抢12306火车票的代码(附源码)
Jan 24 Python
python实现冒泡排序算法的两种方法
Mar 10 Python
Python嵌套列表转一维的方法(压平嵌套列表)
Jul 03 Python
python爬虫获取百度首页内容教学
Dec 23 Python
详解Python解决抓取内容乱码问题(decode和encode解码)
Mar 29 Python
TensorFlow加载模型时出错的解决方式
Feb 06 Python
python 中不同包 类 方法 之间的调用详解
Mar 09 Python
Flask处理Web表单的实现方法
Jan 31 Python
python爬取企查查企业信息之selenium自动模拟登录企查查
Apr 08 Python
如何用六步教会你使用python爬虫爬取数据
Apr 06 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
无数据库的详细域名查询程序PHP版(1)
2006/10/09 PHP
php select,radio和checkbox默认选择的实现方法
2010/05/15 PHP
php中heredoc与nowdoc介绍
2014/12/25 PHP
Zend Framework框架之Zend_Mail实现发送Email邮件验证功能及解决标题乱码的方法
2016/03/21 PHP
PHP defined()函数的使用图文详解
2019/07/20 PHP
javascript css在IE和Firefox中区别分析
2009/02/18 Javascript
Visual Studio中的jQuery智能提示设置方法
2010/03/27 Javascript
jQuery 验证插件 Web前端设计模式(asp.net)
2010/10/17 Javascript
使用Mootools动态添加Css样式表代码,兼容各浏览器
2011/12/12 Javascript
JS正则表达式大全(整理详细且实用)
2013/11/14 Javascript
window.setInterval()方法的定义和用法及offsetLeft与style.left的区别
2015/11/11 Javascript
jQuery移动页面开发中的触摸事件与虚拟鼠标事件简介
2015/12/03 Javascript
jQuery实例—选项卡的简单实现(js源码和jQuery)
2016/06/14 Javascript
jQuery弹出遮罩层效果完整示例
2016/09/13 Javascript
使用BootStrapValidator完成前端输入验证
2016/09/28 Javascript
js正则表达式惰性匹配和贪婪匹配用法分析
2016/12/26 Javascript
AngularJs ng-change事件/指令的用法小结
2017/11/01 Javascript
JS计算距当前时间的时间差实例
2017/12/29 Javascript
Express进阶之log4js实用入门指南
2018/02/10 Javascript
vue webpack打包后图片路径错误的完美解决方法
2018/12/07 Javascript
一篇文章带你搞懂Vue虚拟Dom与diff算法
2020/08/25 Javascript
Python内置函数bin() oct()等实现进制转换
2012/12/30 Python
pyqt4教程之实现半透明的天气预报界面示例
2014/03/02 Python
Python基于pygame实现的弹力球效果(附源码)
2015/11/11 Python
python 写的一个爬虫程序源码
2016/02/28 Python
pandas 快速处理 date_time 日期格式方法
2018/11/12 Python
python通过tcp发送xml报文的方法
2018/12/28 Python
Django中reverse反转并且传递参数的方法
2019/08/06 Python
Python os模块常用方法和属性总结
2020/02/20 Python
浅谈python量化 双均线策略(金叉死叉)
2020/06/03 Python
基于HTML5 Canvas 实现弹出框效果
2017/06/05 HTML / CSS
ASOS西班牙官网:英国在线时尚和美容零售商
2020/01/10 全球购物
品酒会策划方案
2014/05/26 职场文书
新闻稿件写作技巧
2015/07/18 职场文书
大学生暑期社会实践的个人总结!
2019/07/17 职场文书
前端框架ECharts dataset对数据可视化的高级管理
2022/12/24 Javascript