From e9d09e2df9814ad56f6eec61790b2de6283243ae Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 5 Dec 2019 17:05:54 +0100 Subject: randomize class --- cli/app/commands/biggan/random.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'cli/app/commands') diff --git a/cli/app/commands/biggan/random.py b/cli/app/commands/biggan/random.py index ebfde4b..e7c198b 100644 --- a/cli/app/commands/biggan/random.py +++ b/cli/app/commands/biggan/random.py @@ -30,7 +30,7 @@ def cli(ctx): import tensorflow_hub as hub print("Loading module...") - module = hub.Module('https://tfhub.dev/deepmind/biggan-128/2') + module = hub.Module('https://tfhub.dev/deepmind/biggan-512/2') # module = hub.Module('https://tfhub.dev/deepmind/biggan-256/2') # module = hub.Module('https://tfhub.dev/deepmind/biggan-512/2') @@ -39,8 +39,9 @@ def cli(ctx): z = truncation * tf.random.truncated_normal([batch_size, 120]) # noise sample - y_index = tf.random.uniform([batch_size], maxval=1000, dtype=tf.int32) - y = tf.one_hot(y_index, 1000) + # y_index = tf.random.uniform([batch_size], maxval=1000, dtype=tf.int32) + # y = tf.one_hot(y_index, 1000) + y = tf.random.normal([batch_size]) outputs = module(dict(y=y, z=z, truncation=truncation)) -- cgit v1.2.3-70-g09d2