summaryrefslogtreecommitdiff
path: root/cli/app/commands
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-12-11 00:03:41 +0100
committerJules Laplace <julescarbon@gmail.com>2019-12-11 00:03:41 +0100
commit890be0143f8e32e79d91b56a653c01072b532990 (patch)
tree0a4926af8505b9f0da346cf599f7c1bf4f10927b /cli/app/commands
parentf702091a6f54698b03f9eba2702f023dc6358fbd (diff)
steps
Diffstat (limited to 'cli/app/commands')
-rw-r--r--cli/app/commands/biggan/search.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search.py b/cli/app/commands/biggan/search.py
index 9ee440e..0073769 100644
--- a/cli/app/commands/biggan/search.py
+++ b/cli/app/commands/biggan/search.py
@@ -69,7 +69,7 @@ def cli(ctx, opt_fp_in, opt_dims, opt_steps, opt_video, opt_tag):
out_images = out_file.create_dataset('xtrain', (len(paths), 3, 512, 512,), dtype='float32')
out_labels = out_file.create_dataset('ytrain', (len(paths), vocab_size,), dtype='float32')
out_fns = out_file.create_dataset('fn', (len(paths),), dtype=h5py.string_dtype())
- for path, index in enumerate(paths):
+ for index, path in enumerate(paths):
out_fns[index] = os.path.basename(path)
fp_frames = find_nearest_vector(generator, sess, input_z, input_y, input_trunc, output, path, opt_dims, out_images, out_labels, opt_steps, index)
if opt_video: