summaryrefslogtreecommitdiff
path: root/inversion/image_inversion.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-12-11 21:28:34 +0100
committerJules Laplace <julescarbon@gmail.com>2019-12-11 21:28:34 +0100
commitf25796a03ebc54475c10b0c22bb01ae9a413a9b2 (patch)
tree8f775114d0e19aa2985996058a81da8febce3ca5 /inversion/image_inversion.py
parentbc9cb9fc5d5889c7251e9f0b997fddcb97e3d69e (diff)
fix
Diffstat (limited to 'inversion/image_inversion.py')
-rw-r--r--inversion/image_inversion.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py
index b050e86..3c0a4b7 100644
--- a/inversion/image_inversion.py
+++ b/inversion/image_inversion.py
@@ -358,7 +358,7 @@ if COND_GAN:
out_labels = out_file.create_dataset('ytrain', (NUM_IMGS_TO_PROCESS, N_CLASS,), dtype='float32')
out_err = out_file.create_dataset('err', (NUM_IMGS_TO_PROCESS,))
-out_fns[:] = sample_fns
+out_fns[:len(sample_fns)] = sample_fns
# Gradient descent w.r.t. generator's inputs.
it = 0