如何选择密集层中的单位数?

时间:2019-02-28 13:14:52

标签: tensorflow keras neural-network conv-neural-network

我试图了解tensorflow tutorial中密集层中的单位数量如何影响事物。

pool2_flat = tf.reshape(pool2, [-1, 7 * 7 * 64])    
dense = tf.layers.dense(inputs=pool2_flat, units=1024, activation=tf.nn.relu)

这里单位的选择是否与输入pool2_flat的形状有关?这个数字有多重要?如果之后有更多的卷积层,则必须将此密集输出调整为4D批处理图像张量。因此,我猜想在密集层类型中选择单位是命令,随后的卷积层需要做什么才能获得正确的输出尺寸?关于如何选择它有任何指导或直觉吗?

0 个答案:

没有答案