搭建一个简简单单的2D卷积神经网络

最新推荐文章于 2023-09-25 20:34:26 发布

fK0pS

最新推荐文章于 2023-09-25 20:34:26 发布

阅读量331

点赞数

分类专栏： ML 文章标签：神经网络 pytorch 深度学习

本文链接：https://blog.csdn.net/Hodors/article/details/120639345

版权

ML 专栏收录该内容

81 篇文章 1 订阅

订阅专栏

import torch
import torch.nn as nn
import torch.nn.functional as F

#2D卷积神经网络
input = torch.randn(1, 1, 28, 28)

# With square kernels and equal stride
conv1 = nn.Conv2d(1, 10, kernel_size=5)
conv2 = nn.Conv2d(10, 20, kernel_size=5)
conv2_drop = nn.Dropout2d()
fc1 = nn.Linear(320, 50)
fc2 = nn.Linear(50, 10)


X = F.relu(F.max_pool2d(conv1(input),2))
X = F.relu(F.max_pool2d(conv2_drop(conv2(X)),2))
X = X.view(-1, 320)
X = F.relu(fc1(X))
X = F.dropout(X, training=True)
X = fc2(X)
X = F.log_softmax(X, dim=0)


print("output shape is {}".format(X.shape))

=================

1、学会了如何快速调整、改变每一层张量的大小；

2、学会了 搭建网络模型 和 准备训练数据 相互解耦的流程；

3、学会了如何阅读别人网络结构；