update

liuning-scu-cn · Aug 3, 2019 · ccd1575 · ccd1575
1 parent 3a14eb2
commit ccd1575
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -35,9 +35,9 @@ def main():
     logging = init_log(save_dir)
     _print = logging.info
 
-    train_paths = ['./datalist/dir/train.txt', ]
-    gallery_paths = ['./datalist/dir/gallery.txt', ]
-    probe_paths = ['./datalist/dir/probe.txt', ]
+    train_paths = ['./datalist/train.txt', ]
+    gallery_paths = ['./datalist/gallery.txt', ]
+    probe_paths = ['./datalist/probe.txt', ]
 
     train_iter, gallery_iter, probe_iter = load_direction_gallery_probe(
         root='./database',