(step)
| 254 | E.load_weights(checkpoint_dir+"/E_weights.keras") |
| 255 | |
| 256 | def f_save(step): |
| 257 | save(sess, saver, checkpoint_dir, step, name) |
| 258 | G.save_weights(checkpoint_dir+"/G_weights.keras", True) |
| 259 | D.save_weights(checkpoint_dir+"/D_weights.keras", True) |
| 260 | E.save_weights(checkpoint_dir+"/E_weights.keras", True) |
| 261 | |
| 262 | def sampler(z, x): |
| 263 | code = E.predict(x, batch_size=batch_size)[0] |