首页 文章

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

提问于
浏览
3

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

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

任何能够提供一些见解的人都将不胜感激 .

1 回答

  • 4

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

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

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

相关问题