设置权重以从具有附加图层的另一个模型建模

时间:2019-09-27 14:42:40

标签: python tensorflow keras

我正在尝试在Keras模型中更新权重(让它成为模型A)。

权重是从另一个模型(模型B-它是模型A的扩展,最后是两个额外的层)给出的,该模型处于训练循环过程中(train_on_batch),我使用方法来更新权重:

modelA.set_weights(modelB.get_weights())

令人惊讶的是,即使网络的体系结构不同,它也可以工作(过程运行)。那怎么可能set_weights()是否会自动削减其他部分权重?还是有什么问题和更新的重量混淆了?

1 个答案:

答案 0 :(得分:1)

似乎set_weights方法加载了有关图层名称的float列表。

enter image description here

您可能正在寻找带有参数load_weightsdocumentation)的by_name=True方法