我正在使用来自here的名为make_unet
的卷积神经网络(CNN)。它工作正常,代码能够与此CNN一起运行。但我知道,在深度学习中,你必须初始化权重以优化神经网络。
Keras中的documentation清楚地表明使用kernel_initializer
进行权重初始化。但是,我在kernel_initializer
函数中看不到任何make_unet
。
任何能提供一些见解的人都会受到赞赏。
答案 0 :(得分:4)
在Keras中,初始化者通过参数kernel_initializer
和bias_initializer
按层传递,例如。
Dense(64, kernel_initializer='random_uniform', bias_initializer='zeros')
所有内置图层都带有合理的默认初始化程序。例如,所有卷积层都使用kernel_initializer='glorot_uniform'
,bias_initializer='zeros'
。 Keras为您提供了许多alternative options。您还可以创建自定义初始化程序。