diff options
| -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: |