pytorch学习
文章平均质量分 83
Y.F,Ge
11
展开
-
nn.Module模块的分类与使用
前言本文比较了以下几种神经网络模型容器的区别一、容器类型1.有序型代码如下:1.1.LeNetSequentialclass LeNetSequential(nn.Module): def __init__(self, classes): super(LeNetSequential, self).__init__() self.features = nn.Sequential( nn.Conv2d(3, 6, 5),原创 2021-10-27 21:50:24 · 799 阅读 · 0 评论 -
由x = x.view(x.size()[0], -1)引发的对CNN模型的分析
项目场景:提示:自己搭建CNN以及ALexNet时:问题描述:有一段代码看不懂:x = x.view(x.size()[0], -1)初步分析:应该是让特征张量扁平化,一维化。class LeNetSequential(nn.Module): def __init__(self, classes): super(LeNetSequential, self).__init__() self.features = nn.Sequential(原创 2021-10-27 16:32:14 · 432 阅读 · 0 评论