在Keras中使用不同名称的图层加载权重

时间:2018-03-20 12:47:04

标签: python tensorflow keras

我的模型由两个不同的ResNets组成,它的图层共享相同的名称,但在其中一个分支的开头有一个d_前缀。

Image1----conv1---conv2---(...)
                                   \
                                     Other stuff I don't need the weights
                                   /
Image2----d_conv1---d_conv2---(...)

如果我的weights.h5文件的权重仅使用不带前缀的名称保存,那么如何为ResNet加载相同的权重?

1 个答案:

答案 0 :(得分:0)

在Keras中加载由by_name = False设置的权重。