标签: machine-learning neural-network conv-neural-network
我已针对图像分类器问题实现了结合了VGG16 Conv和自定义FC的自定义CNN。我的分类器的输出为2类。
我使用了2个密集层,密集层架构如下所述。
输入25088 => Dense1(1024个节点)=> Dense2(2个节点)
使用这种架构,我的模型大小约为300MB。由于内存限制,我无法运行巨大的模型。因此,我将Dense1 1024个节点减少到128个节点。
这里我有2个问题,
预先感谢。