From 0e6d765dfc3a0fae6c012301c9b5313c20b38559 Mon Sep 17 00:00:00 2001 From: Alexsandro Souza Date: Sun, 5 Jul 2020 17:26:53 +0100 Subject: [PATCH] Fix for incompatible layer name in Collab --- 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 867d2a6d..f75fb812 100755 --- a/encoder/generator_model.py +++ b/encoder/generator_model.py @@ -32,7 +32,7 @@ def __init__(self, model, batch_size, randomize_noise=False): self.dlatent_variable = next(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') + self.generator_output = self.graph.get_tensor_by_name('G_synthesis_1/_Run/concat/concat:0') self.generated_image = tflib.convert_images_to_uint8(self.generator_output, nchw_to_nhwc=True, uint8_cast=False) self.generated_image_uint8 = tf.saturate_cast(self.generated_image, tf.uint8)