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基于pygame实现的font游戏字体(附源码)
Nov 11 Python
Python网络编程之TCP与UDP协议套接字用法示例
Feb 02 Python
Python实现的读写json文件功能示例
Jun 05 Python
python简易远程控制单线程版
Jun 20 Python
windows下python安装小白入门教程
Sep 18 Python
Python实现的IP端口扫描工具类示例
Feb 15 Python
pytorch nn.Conv2d()中的padding以及输出大小方式
Jan 10 Python
PyTorch: Softmax多分类实战操作
Jul 07 Python
Python getattr()函数使用方法代码实例
Aug 10 Python
详解python 内存优化
Aug 17 Python
python 实现Requests发送带cookies的请求
Feb 08 Python
Django基础CBV装饰器和中间件
Mar 22 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:风雨欲来 路在何方?
2006/10/09 PHP
php中print(),print_r(),echo()的区别详解
2014/12/01 PHP
简单JS代码压缩器
2006/10/12 Javascript
国外的为初学者写的JavaScript教程
2008/06/09 Javascript
javascript 尚未实现错误解决办法
2008/11/27 Javascript
InnerHtml和InnerText的区别分析
2009/03/13 Javascript
清空上传控件input file的值
2010/07/03 Javascript
jQuery之选项卡的简单实现
2014/02/28 Javascript
Nodejs实现的一个静态服务器实例
2014/12/06 NodeJs
高性能JavaScript模板引擎实现原理详解
2015/02/05 Javascript
AngularJS中如何使用$http对MongoLab数据表进行增删改查
2016/01/23 Javascript
解析JavaScript中的字符串类型与字符编码支持
2016/06/24 Javascript
jQuery的图片轮播插件PgwSlideshow使用详解
2016/08/11 Javascript
Angular 2应用的8个主要构造块有哪些
2016/10/17 Javascript
jQuery事件对象总结
2016/10/17 Javascript
AngularJS框架的ng-app指令与自动加载实现方法分析
2017/01/04 Javascript
JS中的Replace()传入函数时的用法详解
2017/09/11 Javascript
解决LayUI表单获取不到data的问题
2018/08/20 Javascript
详解Vue一个案例引发「内容分发slot」的最全总结
2018/12/02 Javascript
使用 vue 实例更好的监听事件及vue实例的方法
2019/04/22 Javascript
Python def函数的定义、使用及参数传递实现代码
2014/08/10 Python
Python浅拷贝与深拷贝用法实例
2015/05/09 Python
详解python中requirements.txt的一切
2017/03/03 Python
python编写简易聊天室实现局域网内聊天功能
2018/07/28 Python
详解Python安装tesserocr遇到的各种问题及解决办法
2019/03/07 Python
基于Numpy.convolve使用Python实现滑动平均滤波的思路详解
2019/05/16 Python
Python学习笔记之Break和Continue用法分析
2019/08/14 Python
python3的数据类型及数据类型转换实例详解
2019/08/20 Python
python实现根据给定坐标点生成多边形mask的例子
2020/02/18 Python
解决jupyter notebook 前面书写后面内容消失的问题
2020/04/13 Python
加拿大消费电子和手机购物网站:The Source
2017/01/28 全球购物
卡西欧G-SHOCK英国官网: 防水防震手表
2018/01/08 全球购物
应届电子商务毕业自荐书范文
2014/02/11 职场文书
实习生矿工检讨书
2014/10/13 职场文书
聘任合同书
2015/09/21 职场文书
思想品德课教学反思
2016/02/24 职场文书