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"))