Python深度学习pytorch神经网络多层感知机简洁实现
作者：Supre_yuan 发布时间：2021-08-20 17:37:43
标签：pytorch,神经网络,感知机
我们可以通过高级API更简洁地实现多层感知机。
import torch
from torch import nn
from d2l import torch as d2l
模型
与softmax回归的简洁实现相比，唯一的区别是我们添加了2个全连接层。第一层是隐藏层，它包含256个隐藏单元，并使用了ReLU激活函数。第二层是输出层。
net = nn.Sequential(nn.Flatten(),
nn.Linear(784, 256),
nn.ReLU(),
nn.Linear(256, 10))
def init_weights(m):
if type(m) == nn.Linear:
nn.init.normal_(m.weight, std=0.01)
net.apply(init_weights)
训练过程的实现与我们实现softmax回归时完全相同，这种模块化设计使我们能够将与和模型架构有关的内容独立出来。
batch_size, lr, num_epochs = 256, 0.1, 10 # 批量大小为256，学习率为0.1，类型为10
loss = nn.CrossEntropyLoss() # 使用交叉熵损失函数
trainer = torch.optim.SGD(net.parameters(), lr=lr) # 开始训练
train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size)
d2l.train_ch3(net, train_iter, test_iter, loss, num_epochs, trainer)
以上就是Python深度学习pytorch神经网络多层感知机简洁实现的详细内容，更多关于pytorch神经网络的资料请关注脚本之家其它相关文章！
来源：https://blog.csdn.net/weixin_43880225/article/details/120614502
0
投稿猜你喜欢
- 本文实例讲述了php实现的验证码文件类。分享给大家供大家参考。具体如下：<?php/*** @file* @version 1.0*
使用图层可以像素为单位精确定位页面元素，并且可以将层放置在页面的任意位置。当把页面元素放入图层之中时，还可以控制哪个显示在前面、哪个显示在后- 本文实例讲述了Python图像处理之图像的缩放、旋转与翻转实现方法。分享给大家供大家参考，具体如下：图像的几何变换，如缩放、旋转和翻转等，在
- 本文介绍基于Python语言arcpy模块，实现栅格影像图层建立与多幅遥感影像数据批量拼接（Mosaic）的操作。首先，相关操作所需具体代码
- Pandas中根据列的值选取多行数据 # 选取等于某些值的行记录 用 == df.loc[df['column_name']
- master库对于SQLServer来说，是很重要的系统数据库，保存着所有Sqlserver的用户信息、数据库信息等，当数据库崩溃时，mas
- 1下载安装1.1打开官网http://www.jetbrains.com/pycharm/download/#section=windows
- 一、简单介绍pip 是 Python 包管理工具，该工具提供了对Python 包的查找、下载、安装和卸载的功能，现在大家用到的所有包不是自带
- 一、牛顿多项式拉格朗日多项式的公式不具备递推性，每个多项式需要单独构造。但很多时候我们需要从若干个逼近多项式选择一个。这个时候我们就需要一个
- 这篇文章主要介绍了python 采用paramiko 远程执行命令及报错解决,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的
- 在mysql中带了随机取数据的函数，在mysql中我们会有rand()函数，很多朋友都会直接使用，如果几百条数据肯定没事，如果几万或百万时你
- vue 百度地图 + 定位 前提需要自己有百度的密钥，如没有可以去百度地图申请一、在主目录下的index.html引入js，例如：
- 如下所示：beta分布的最大特点是其多样性, 从下图可以看出, beta分布具有各种形态, 有U形, 类似正态分布的形状, 类似unifor
- 前言pickle 模块可以对一个 Python 对象的二进制进行序列化和反序列化。说白了，就是它能够实现任意对象与二进制直接的相互转化，也可
- python调用文件时找不到相对路径current_path = os.path.dirname(__file__)if current_p
- 效果图:基于d3-v5, 依赖dagre-d3, 直接上代码:<!DOCTYPE html><html lang=&quo
- pytorch 中的 state_dict 是一个简单的python的字典对象,将每一层与它的对应参数建立映射关系.(如model的每一层的
- Flask是一个用Python编写的Web应用程序框架，Flask是python的web框架，最大的特征是轻便，让开发者自由灵活的兼容要开发
- 引言本文讨论 ChatGTP 与 Node.js 的结合，由于最近在学 Remix, 考虑到 Remix 可以单独的作为 API 来使用，并
- 为了更好的说明问题，首先引出下面的题目//请说明下面变量 a-d 的值 var a = [[1][1]]; var b = [['a
