Keras-GAN icon indicating copy to clipboard operation
Keras-GAN copied to clipboard

Model saving and testing on external data

Open hemanthkumar3111 opened this issue 5 years ago • 3 comments

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

hemanthkumar3111 avatar May 31 '20 09:05 hemanthkumar3111

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.

hiteshnitetc avatar Jun 12 '20 13:06 hiteshnitetc

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

KamiKazeFei avatar Jul 11 '20 14:07 KamiKazeFei

Just use load_model, and then train.

On Sat, Jul 11, 2020 at 8:29 PM KamiKazeFei [email protected] wrote:

@hiteshnitetc https://github.com/hiteshnitetc after I save the weight and model, how to do can use those model and weight to re-train? thx

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/eriklindernoren/Keras-GAN/issues/222#issuecomment-657076609, or unsubscribe https://github.com/notifications/unsubscribe-auth/AHJSSSUUKFKSFRDWRIB6T53R3B46VANCNFSM4NPCY2UQ .

-- Hitesh Tekchandani PhD Research Scholar NIT Raipur India

hiteshnitetc avatar Jul 11 '20 15:07 hiteshnitetc