【发布时间】: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