Generative Adversarial Networks Projects
上QQ阅读APP看书,第一时间看更新

Saving the models

After training is complete, save the learned weights of the generator and the discriminator models by adding the following code:

"""
Save models
"""
generator.save_weights(os.path.join(generated_volumes_dir, "generator_weights.h5"))
discriminator.save_weights(os.path.join(generated_volumes_dir, "discriminator_weights.h5"))