From 7699a5ef2a27393a5fa55503d4d54c1ce66241f9 Mon Sep 17 00:00:00 2001
From: Uha661 <54659541+Uha661@users.noreply.github.com>
Date: Tue, 3 Sep 2019 11:58:11 +0530
Subject: [PATCH] Update generator_model.py

---
 encoder/generator_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/encoder/generator_model.py b/encoder/generator_model.py
index 867d2a6d6..8ffd805dd 100644
--- a/encoder/generator_model.py
+++ b/encoder/generator_model.py
@@ -29,7 +29,7 @@ def __init__(self, model, batch_size, randomize_noise=False):
         self.sess = tf.get_default_session()
         self.graph = tf.get_default_graph()
 
-        self.dlatent_variable = next(v for v in tf.global_variables() if 'learnable_dlatents' in v.name)
+        self.dlatent_variable = (v for v in tf.global_variables() if 'learnable_dlatents' in v.name)
         self.set_dlatents(self.initial_dlatents)
 
         self.generator_output = self.graph.get_tensor_by_name('G_synthesis_1/_Run/concat:0')