【一起深度学习——NIN】

ops/2024/10/10 17:53:48/

NIN神经网络

  • 原理图:
  • 代码实现:
  • 输出结果:

原理图:

请添加图片描述

代码实现:

import torch
from torch import nn
from d2l import torch as d2ldef nin_block(in_channels, out_channels, kernel_size, strides, padding):return nn.Sequential(#模块化设计思想,方便重复使用。#经过一个卷积层,再加两个1*1的卷积层。起到全连接层的作用。nn.Conv2d(in_channels,out_channels,kernel_size,strides,padding),nn.ReLU(),# 两个1*1的卷积层,并不会改变形状,不改变通道数nn.Conv2d(out_channels, out_channels, kernel_size=1, stride=1),nn.ReLU(),nn.Conv2d(out_channels,out_channels,kernel_size=1,stride=1),nn.ReLU(),)#nn.Sequential 容器
net = nn.Sequential(nin_block(1,96,kernel_size=11,strides=4,padding=0),nn.MaxPool2d(kernel_size=3,stride=2),nin_block(96,256,kernel_size=5,strides=1,padding=2),nn.MaxPool2d(kernel_size=3,stride=2),nin_block(256,384,kernel_size=3,strides=1,padding=1),nn.MaxPool2d(kernel_size=3,stride=2),nn.Dropout(0.5),# [1,10,5,5]nin_block(384, 10, kernel_size=3, strides=1, padding=1),#自适应平均池化。与普通的池化层不同,允许指定输出特征图的大小而不是池化层的大小。#[1,10,5,5] => [1,10,1,1]nn.AdaptiveAvgPool2d(output_size=(1,1)),# [1,10,1,1] =>[1,10*1*1] =>[1,10]nn.Flatten()
)#开始训练
lr, num_epochs, batch_size = 0.1, 10, 128
train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size, resize=224)
d2l.train_ch6(net, train_iter, test_iter, num_epochs, lr, d2l.try_gpu())

输出结果:

请添加图片描述
请添加图片描述


http://www.ppmy.cn/ops/35959.html

相关文章

模拟实现小米商城(有源码)

项目介绍 这是一个只有前端没有后端的项目, 适合于基础前端课设.该前端项目没有花里胡哨的特效, 纯手写 HTML CSS JS, 特别适合基础小白入门, 或者做为基础的课设(含有组员介绍页面), 此项目中, 充分使用 flex 布局, 绝对 相对定位, css 动画, 封装公共样式区域, 代码风格及命…

Java 笔记 08:Scanner 类的使用方法

一、前言 记录时间 [2024-04-28] 系列文章简摘: Java 笔记 01:Java 概述,MarkDown 常用语法整理 Java 笔记 04:Java 数据类型基础,数据类型转换,及其相关场景拓展 Java 笔记 05:变量和常量相关知…

SSL证书中DV、OV和EV有什么区别,又该如何选择

SSL(安全套接层)证书作为一种加密工具,确保了网站与其用户之间传输的信息的安全性。而在选择SSL证书时,我们通常会看到三种类型:域名验证(DV)、组织验证(OV)和扩展验证&a…

【一起深度学习——批量规范化】

批量规范化 1、为啥要批量规范化呢?2、如何批量规范化呢?3、实现批量归一化。4、定义BatchNorm层:5、定义神经网络:6、开始训练: 1、为啥要批量规范化呢? 1、可持续加速深层网络的收敛速度。 2、对于深层网…

【Python小技巧】matplotlib不显示图像竟是numpy惹的祸

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 文章目录 前言一、问题:df.plot() 显示不出图像二、尝试各种解决办法1. 增加matplotlib.use,设定GUI2. 升级matplotlib版本 三、numpy是个重要的库1. …

iPhone查看本机号码只需要这3招,不再为号码忘记犯愁!

在日常生活中,我们经常需要使用手机号码进行各种通讯活动,但有时候会忘记自己的手机号码,让人感到非常尴尬。不过,如果您是iPhone用户,那么您可以放心了!因为在iphone查看本机号码只需要简单的几个步骤&…

检测服务器环境,实现快速部署。适用于CRMEB_PRO/多店

运行效果如图: 最近被好多人问,本来运行的好好的,突然swoole就启动不了了。 本工具为爱发电,如果工具正好解决了您的需求。我会很开心 代码如下: """本脚本为爱发电by:网前雨刮器 """…

2024年美国市场亚太游戏品牌数字广告洞察报告

来源:Sensor Tower 美国是全球最大的游戏市场之一,也是亚太游戏品牌出海的重要市场。2023年Q2至2024年Q1,美国市​场广告投放额排名前10的亚太游戏品牌,合计支出 超过7.5亿美元,环比上涨23%。 排名第一的米哈游(miHoY…