【问题标题】:Can we get weights by name in keras?我们可以在 keras 中按名称获取权重吗?
【发布时间】:2021-08-31 04:05:32
【问题描述】:

通过model.get_layer("layer_name")可以得到模型model的图层对象。

是否有任何 keras 函数可以做类似于权重的事情?

用法类似于 model.get_layer("layer_name").get_variable_by_name("recurrent_kernel")

在使用以下定义的循环权重构建的层中:

self.recurrent_kernel = self.add_weight(shape=(self.units, self.units),
        initializer=self.recurrent_initializer,
        name="recurrent_kernel", trainable=self.train_recurrent)

编辑:这不是我正在寻找的解决方案:Why get_tensor_by_name can't get the weights of layers defined by tf.keras.layers properly

【问题讨论】:

    标签: tensorflow keras


    【解决方案1】:

    这并不理想,但虽然没有答案。我正在使用这个:

    def get_weights_by_name(model, name):
        return [w for w in model.weights if w.name==name][0]
    

    【讨论】:

      猜你喜欢
      • 2023-03-16
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-05-16
      • 2017-03-19
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多