tensorflow 2.0 中多个模型以上的保存,转移以及 断续训练模型的方法以及tf.keras.callbacks.ModelCheckpoint无效的原因

在学习tensorflow2.0过程中,查阅了很多关于模型的保存以及转移的方法,其中看到过最多的就是tf.keras.callbacks.ModelCheckpoint的使用。但是实际上测试的过程中通过此方法并不能生成任何模型的保存文件,其他的方法均是在compile,fit的demo下做的,因此如果一般的自建模型应该如何去保存和重载并进行断点继续训练呢?因此自己摸索,实验结果如下。

实验代码是WGAN,因为涉及到generator和discriminator,所以我把这个模型作为两个模型的测试代码。即如何同时恢复generator和discriminator两个模型的参数。

全文请打赏2元后,发(支付截图和你的邮箱地址)到[email protected],我会立即发送全文。总结不易,望请理解。
tensorflow 2.0 中多个模型以上的保存,转移以及 断续训练模型的方法以及tf.keras.callbacks.ModelCheckpoint无效的原因