diff options
| -rw-r--r-- | inversion/image_inversion.py | 2 |
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 |