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的动态重新封装的教程
Apr 11 Python
Python编程中用close()方法关闭文件的教程
May 24 Python
Django中URLconf和include()的协同工作方法
Jul 20 Python
Python创建xml文件示例
Mar 22 Python
python框架中flask知识点总结
Aug 17 Python
python判断文件是否存在,不存在就创建一个的实例
Feb 18 Python
Python爬虫实现验证码登录代码实例
May 10 Python
python numpy 常用随机数的产生方法的实现
Aug 21 Python
python 实现按对象传值
Dec 26 Python
Pytorch 实现数据集自定义读取
Jan 18 Python
tensorflow实现在函数中用tf.Print输出中间值
Jan 21 Python
Python基础之操作MySQL数据库
May 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
德生BCL3000的电路分析和打磨
2021/03/02 无线电
Http 1.1 Etag 与 Last-Modified提高php效率
2008/01/10 PHP
PHP实现基于图的深度优先遍历输出1,2,3...n的全排列功能
2017/11/10 PHP
ajax+php实现无刷新验证手机号的实例
2017/12/22 PHP
浅谈Laravel POST,PUT,PATCH 路由的区别
2019/10/15 PHP
图片按比例缩放函数
2006/06/26 Javascript
JavaScript中的其他对象
2008/01/16 Javascript
HTML node相关的一些资料整理
2010/01/01 Javascript
编写可维护面向对象的JavaScript代码[翻译]
2011/02/12 Javascript
JS获取页面input控件中所有text控件并追加样式属性
2013/02/25 Javascript
最流行的Node.js精简型和全栈型开发框架介绍
2015/02/26 Javascript
jQuery异步上传文件插件ajaxFileUpload详细介绍
2015/05/19 Javascript
三种Node.js写文件的方式
2016/03/08 Javascript
js实现手机发送验证码功能
2017/03/13 Javascript
node+vue实现用户注册和头像上传的实例代码
2017/07/20 Javascript
vue+axios+element ui 实现全局loading加载示例
2018/09/11 Javascript
JS编写兼容IE6,7,8浏览器无缝自动轮播
2018/10/12 Javascript
vuejs简单验证码功能完整示例
2019/01/08 Javascript
详解关于html,css,js三者的加载顺序问题
2019/04/10 Javascript
axios+Vue实现上传文件显示进度功能
2019/04/14 Javascript
React组件对子组件children进行加强的方法
2019/06/23 Javascript
Python实现读取目录所有文件的文件名并保存到txt文件代码
2014/11/22 Python
在Python的Flask框架中使用日期和时间的教程
2015/04/21 Python
Python使用Shelve保存对象方法总结
2019/01/28 Python
python中字符串数组逆序排列方法总结
2019/06/23 Python
python实现函数极小值
2019/07/10 Python
django 信号调度机制详解
2019/07/19 Python
使用PyQt5实现图片查看器的示例代码
2020/04/21 Python
Python Json数据文件操作原理解析
2020/05/09 Python
python如何调用字典的key
2020/05/25 Python
高校学生干部的自我评价分享
2013/11/04 职场文书
学校运动会广播稿范文
2014/10/02 职场文书
暑假生活随笔
2015/08/15 职场文书
考研经验交流会策划书
2015/11/02 职场文书
Python爬虫之爬取哔哩哔哩热门视频排行榜
2021/04/28 Python
vue2实现provide inject传递响应式
2021/05/21 Vue.js