pytorch学习(六)---搭建简单的神经网络以及sequential的使用

2023-11-11

        本篇自学笔记来自于b站《PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】》,Up主讲的非常通俗易懂,文章下方有视频连接,如有需要可移步up主讲解视频,如有侵权,实非故意,深表歉意,请与我联系,删除相关内容!

        本节以CIFAR10的模型结果(如下图)为例,搭建该简易的神经网络。

         首先需要知道某一个尺寸的图片经过卷积之后的尺寸如何计算。

        知道怎么计算之后,看模型图输入为32*32,经过一个5*5卷积核卷积之后,图片的尺寸依旧是32*32,通过上述公式,W1=32

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

pytorch学习(六)---搭建简单的神经网络以及sequential的使用 的相关文章