remainder

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))