summaryrefslogtreecommitdiff
path: root/inversion
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-12-11 21:33:36 +0100
committerJules Laplace <julescarbon@gmail.com>2019-12-11 21:33:36 +0100
commit32243e95205e6704b0afa60f5d25e12136ce95ce (patch)
tree65cc5a16c5f67e5a512b0bea46a41dbf9435993b /inversion
parent9d5f67d45e52858840b05ce081cda69bfa7002ad (diff)
fixxx
Diffstat (limited to 'inversion')
-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 7a0ef1e..47703d6 100644
--- a/inversion/image_inversion.py
+++ b/inversion/image_inversion.py
@@ -359,7 +359,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[:len(sample_fns)] = sample_fns
+out_fns[:] = sample_fns[:NUM_IMGS_TO_PROCESS]
# Gradient descent w.r.t. generator's inputs.
it = 0