diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 16:20:46 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 16:20:51 +0100 |
| commit | 4e7c8b0c30cda9cb4e7e991ad9d1313b4f8c9d6f (patch) | |
| tree | a0f36c635d33dff4547effb228b6a8efc19a3001 | |
| parent | 5b270cddfda2da3f583af538d3fc86ff6690f82d (diff) | |
next
| -rw-r--r-- | inversion/image_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 43eb207..a4cec7f 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -367,7 +367,7 @@ for image_batch, label_batch in image_gen: # Initialize encodings to random values. if params.pre_trained_latent: - sess.run(latent.assign(latent_gen.next())) + sess.run(latent.assign(next(latent_gen))) if params.inv_layer != 'latent': sess.run(encoding.assign(gen_encoding)) else: |