tensorflow - Tensorflow 2 部分恢复检查点

Question

我有一个 MobileNetV2 模型，它有 28 个输出类（模型 1）。我还有另一个问题，它有 2 个输出类（模型 2）。我想使用模型 1 的权重来初始化模型 2，因为它们是一个类似的问题（例如迁移学习）。我不能只使用标准恢复，因为最终输出层权重不兼容。

有没有办法部分恢复模型，以便它忽略不兼容的权重，或者不太理想的是，手动按重量进行加权。

谢谢！