change to val

leohsuofnthu · Mar 31, 2020 · 4a926d1 · 4a926d1
1 parent a9dba28
commit 4a926d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_detection.py b/test/test_detection.py
@@ -41,7 +41,7 @@
 # Todo: Figure out why batch size = 1 cause memory issue
 valid_dataset = dataset_coco.prepare_dataloader(tfrecord_dir="../data/coco",
                                                 batch_size=1,
-                                                subset='train')
+                                                subset='val')
 anchors = anchorobj.get_anchors()
 tf.print(tf.shape(anchors))