Tensorflow v2 / Keras。将权重从一种模型转移到另一种模型

时间:2020-09-12 13:21:32

标签: tensorflow keras

我有一个用model.save_weights保存的张量流模型(模型1),我当然可以用model.load_weights还原整个模型。

如果我更改架构(模型2),仍如何从模型1加载编码器并将其传输到模型2?设置by_name = True和skip_mismatch = True仍然会产生

ValueError: Shapes (1, 1, 16, 3) and (1, 1, 16, 2) are incompatible

使用tf.train.load_checkpoint似乎是可能的,但不仅变量名有些奇怪(名称中包含.ATTRIBUTES / VARIABLE_VALUE),而且似乎也不太理想。

有人知道如何以优雅的方式进行转学吗?

最诚挚的问候

0 个答案:

没有答案