From 4e7c8b0c30cda9cb4e7e991ad9d1313b4f8c9d6f Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 11 Dec 2019 16:20:46 +0100 Subject: next --- inversion/image_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'inversion/image_inversion.py') 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: -- cgit v1.2.3-70-g09d2