randomize class

1 files changed, 4 insertions, 3 deletions

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))