如何选择完全连接层中的节点数?

时间:2019-07-05 08:49:08

标签: machine-learning neural-network conv-neural-network

我已针对图像分类器问题实现了结合了VGG16 Conv和自定义FC的自定义CNN。我的分类器的输出为2类。

我使用了2个密集层,密集层架构如下所述。

输入25088 => Dense1(1024个节点)=> Dense2(2个节点)

使用这种架构,我的模型大小约为300MB。由于内存限制,我无法运行巨大的模型。因此,我将Dense1 1024个节点减少到128个节点。

这里我有2个问题,

  1. 这是减少解决方案中节点数量的正确方法吗?
  2. 如何在密集层中选择数字节点?

预先感谢。

0 个答案:

没有答案