Minor fix

yztxwd · yztxwd · commit 3a46a1bf1c6a · 2022-06-21T20:12:31.000-04:00
diff --git a/trainNN/scan_genome.py b/trainNN/scan_genome.py
@@ -16,7 +16,7 @@
 
 def TFdataset(path, batchsize, dataflag):
 
-    TFdataset_batched = iterutils.train_TFRecord_dataset(path, batchsize, dataflag, shuffle=False)
+    TFdataset_batched = iterutils.train_TFRecord_dataset(path, batchsize, dataflag, shuffle=False, drop_remainder=False)
 
     return TFdataset_batched
 
