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

问题描述 投票:0回答:1

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

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

modelA.set_weights(modelB.get_weights())

而且令人惊讶的是,即使网络的体系结构不同,它也能正常工作(过程运行)。那怎么可能set_weights()是否会自动削减其他部分权重?还是出了点问题,更新的重量混合在一起了?

python tensorflow keras
1个回答
0
投票

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

enter image description here

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

© www.soinside.com 2019 - 2024. All rights reserved.