我的CNN模型中的内核权重初始化在哪里?

时间:2018-01-26 07:18:21

标签: python machine-learning deep-learning keras conv-neural-network

我正在使用来自here的名为make_unet的卷积神经网络(CNN)。它工作正常,代码能够与此CNN一起运行。但我知道,在深度学习中,你必须初始化权重以优化神经网络。

Keras中的documentation清楚地表明使用kernel_initializer进行权重初始化。但是,我在kernel_initializer函数中看不到任何make_unet

任何能提供一些见解的人都会受到赞赏。

1 个答案:

答案 0 :(得分:4)

在Keras中,初始化者通过参数kernel_initializerbias_initializer按层传递,例如。

Dense(64, kernel_initializer='random_uniform', bias_initializer='zeros')

所有内置图层都带有合理的默认初始化程序。例如,所有卷积层都使用kernel_initializer='glorot_uniform'bias_initializer='zeros'。 Keras为您提供了许多alternative options。您还可以创建自定义初始化程序。