在发送到Loss Function之前，如何处理CNN层末端的可变大小输入

Question

我正在处理CNN和的可变大小输入，我想知道如何将其馈送到最后一个FC层以满足CrossEntropy Loss函数的要求． 即使对一个样本进行了处理，后续的样本也会有不同的维度，不能用于反向传播。所以我想知道一种或不同的方法来处理这个问题。

(p。S:当前正在使用裁剪输入以使其固定大小，查询是用于即兴创作)

Answer 1

将torch.nn.AdaptiveAvgPool2d(S)放在最后一个转换层和第一个完全连接层之间。

注意，你的全连通层的输入维度应该是S x S x no. of channels in last conv layer