diff options
| author | jules@lens <julescarbon@gmail.com> | 2019-12-11 17:47:58 +0100 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2019-12-11 17:47:58 +0100 |
| commit | 121e283f88d09930e82045bf9b040ea07f9d21f5 (patch) | |
| tree | 4984dee8b77fca8b17129e8a9d63bbac30eed988 /inversion/image_inversion.py | |
| parent | 24b0f264bc85d443ee686cbcd794ca0066a38cbb (diff) | |
| parent | 2a9fbde05922ba0a580546c822942ca77de00e85 (diff) | |
Merge branch 'master' of asdf.us:biggan
Diffstat (limited to 'inversion/image_inversion.py')
| -rw-r--r-- | inversion/image_inversion.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 2c182c1..274353b 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -328,9 +328,9 @@ if params.dataset.endswith('.hdf5'): if NUM_IMGS % BATCH_SIZE != 0: REMAINDER = BATCH_SIZE - (NUM_IMGS % BATCH_SIZE) NUM_IMGS += REMAINDER - sample_images += sample_images[:-REMAINDER] - sample_labels += sample_labels[:-REMAINDER] - sample_fns += sample_fns[:-REMAINDER] + sample_images += sample_images[-REMAINDER:] + sample_labels += sample_labels[-REMAINDER:] + sample_fns += sample_fns[-REMAINDER:] assert(NUM_IMGS % BATCH_SIZE == 0) else: sys.exit('Unknown dataset {}.'.format(params.dataset)) |