eriklindernoren/Keras-GAN

Model saving and testing on external data

hemanthkumar3111 opened this issue · 3 comments

hemanthkumar3111 commented

hi how can I save the model for pix2pix and test the same on my personal data for

hiteshnitetc commented

In def sample_images:
self.discriminator.save("location/dweights_%d.hdf5" %epoch) (for discriminator)
self.generator.save("location/gweights_%d.hdf5" %epoch) (for generator)

you can also save model in JSON, and use save_weights for saving weights.

KamiKazeFei commented

@hiteshnitetc
after I save the weight and model, how to do can use those model and weight to re-train? thx

hiteshnitetc commented