【问题标题】:How to re-initialize layer weights of an existing model in Keras?如何在 Keras 中重新初始化现有模型的层权重?
【发布时间】:2019-10-30 23:31:35
【问题描述】:

实际的问题是在 Keras 中为现有(已经构建的)模型生成随机层权重。有一些使用 Numpy [2] 的解决方案,但选择这些解决方案并不好。因为,在 Keras 中,有特殊的初始化器为每种层类型使用不同的分布。当使用 Numpy 代替初始化器时,生成的权重与原始权重具有不同的分布。举个例子:

我的模型的第二层是卷积 (1D) 层,它的初始化器是 GlorotUniform [1]。如果使用 Numpy 生成随机权重,则生成的权重分布不会是 GlorotUniform。

我有这个问题的解决方案,但它有一些问题。这是我所拥有的:

def set_random_weights(self, tokenizer, config):
    temp_model = build_model(tokenizer, config)
    self.model.set_weights(temp_model.get_weights())

我正在构建现有模型。在构建过程之后,模型的权重被重新初始化。然后我得到重新初始化的权重并将它们设置为另一个模型。生成新权重的构建模型具有冗余过程。所以,我需要一个无需构建模型和 Numpy 的新解决方案。

  1. https://keras.io/initializers/
  2. https://www.codementor.io/nitinsurya/how-to-re-initialize-keras-model-weights-et41zre2g

【问题讨论】:

    标签: tensorflow keras conv-neural-network


    【解决方案1】:

    查看此问题的先前答案here。 具体来说,如果您想使用 Keras 层的原始权重初始化器,您可以执行以下操作:

    import tensorflow as tf
    import keras.backend as K
    
    def init_layer(layer):
        session = K.get_session()
        weights_initializer = tf.variables_initializer(layer.weights)
        session.run(weights_initializer)
    
    
    layer = model.get_layer('conv2d_1')
    init_layer(layer)
    

    【讨论】:

    • 遇到错误:*** AttributeError: 'GlorotUniform' object has no attribute 'run'
    • 你是对的,这个语法不再被支持。在我的答案中查看我更新的代码。
    猜你喜欢
    • 2021-04-23
    • 1970-01-01
    • 2019-04-14
    • 1970-01-01
    • 2017-10-04
    • 2018-11-09
    • 2018-06-20
    • 2021-04-12
    • 1970-01-01
    相关资源
    最近更新 更多