python - Keras：如何保存具有优化器状态的 GAN 模型？

Question

我知道一种用途是使用 Checkpoint 和 CheckpointManager tensorflow 类来保存和恢复 GAN 的训练。

我正在寻找一种方法来保存生成器 (G) 、鉴别器 (D) 及其优化状态，以便我可以操纵模型权重（对于 G 和 D）。对我来说，一个用例是在 GAN 的联邦学习中平均不同生成器的权重。

我发现使用 Keras 或 tensorflow 无法做到这一点。