self.fc = nn.Linear(512, num_classes)这里512为什么不用乘上特征图大小4*4啊

首页课程实战体系课手记专栏慕课教程

self.fc = nn.Linear(512, num_classes)这里512为什么不用乘上特征图大小4*4啊

来源：6-10 PyTorch搭建cifar10训练脚本搭建-ResNet结构（下）

精慕门6239241

2021-09-16

写回答

1回答

会写代码的好厨师

会写代码的好厨师

2021-09-25

512是上一层的纬度的长度。liner层的前一层不能直接用4*4这样的层，要把tensor的shape平铺成向量。

0

0

PyTorch入门到进阶实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发，快速掌握PyTorch框架

1190 学习 · 298 问题

相似问题

embed_size

回答 1

_, pred = output.max(1)

回答 1

为什么numpy数组不可写？

回答 1

在大图中搜索目标图有推荐的模型吗？

回答 1

回答 1

打开慕课网App查看更多内容