diff --git a/tools/train.py b/tools/train.py index a12610f72ffd5765eae5f3b12d452c7fc523c0d4..d0e4b060651507d8c2031fb1b4ed88408fc1a5c1 100644 --- a/tools/train.py +++ b/tools/train.py @@ -1,7 +1,4 @@ from __future__ import division -import sys -sys.path.insert(0, '/mnt/lustre/pangjiangmiao/codebase/mmcv') -sys.path.insert(0, '/mnt/lustre/pangjiangmiao/codebase/mmdet') import argparse from mmcv import Config @@ -17,7 +14,7 @@ def parse_args(): parser = argparse.ArgumentParser(description='Train a detector') parser.add_argument('config', help='train config file path') parser.add_argument('--work_dir', help='the dir to save logs and models') - parser.add_argument('--resume_from', help='the checkpoint to resume from') + parser.add_argument('--resume_from', help='the checkpoint file to resume from') parser.add_argument( '--validate', action='store_true',