channel path

wuwuwuyu · Nov 25, 2020 · 7679419 · 7679419
1 parent 59d6292
commit 7679419
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_object_detection/faster_rcnn/train_res50_fpn.py b/pytorch_object_detection/faster_rcnn/train_res50_fpn.py
@@ -143,7 +143,7 @@ def main(parser_data):
     # 训练设备类型
     parser.add_argument('--device', default='cuda:0', help='device')
     # 训练数据集的根目录
-    parser.add_argument('--data-path', default='../', help='dataset')
+    parser.add_argument('--data-path', default='./', help='dataset')
     # 文件保存地址
     parser.add_argument('--output-dir', default='./save_weights', help='path where to save')
     # 若需要接着上次训练，则指定上次训练保存权重文件地址