From f25796a03ebc54475c10b0c22bb01ae9a413a9b2 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 11 Dec 2019 21:28:34 +0100
Subject: fix

---
 inversion/image_inversion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'inversion/image_inversion.py')

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
-- 
cgit v1.2.3-70-g09d2