diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-05 17:10:22 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-05 17:10:22 +0100 |
| commit | 7f60e705f230e005f033c82ddfeb0261db70d645 (patch) | |
| tree | 655410a0f8ba76a1945735c3b5d17ec15baae8f4 /cli/app/commands | |
| parent | e9d09e2df9814ad56f6eec61790b2de6283243ae (diff) | |
randomize class
Diffstat (limited to 'cli/app/commands')
| -rw-r--r-- | cli/app/commands/biggan/random.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/random.py b/cli/app/commands/biggan/random.py index e7c198b..3e4bff6 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-512/2') + module = hub.Module('https://tfhub.dev/deepmind/biggan-256/2') # module = hub.Module('https://tfhub.dev/deepmind/biggan-256/2') # module = hub.Module('https://tfhub.dev/deepmind/biggan-512/2') @@ -41,7 +41,7 @@ def cli(ctx): # 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]) + y = tf.random.normal([None, 1000]) outputs = module(dict(y=y, z=z, truncation=truncation)) |