From 890be0143f8e32e79d91b56a653c01072b532990 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 11 Dec 2019 00:03:41 +0100 Subject: steps --- cli/app/commands/biggan/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli/app/commands') 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: -- cgit v1.2.3-70-g09d2