Skip to content

Commit 895747d

Browse files
authored
Merge pull request jwyang#158 from severian/fix_argument
Fix save_dir command line option
2 parents 8e029c5 + 89fc25c commit 895747d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trainval_net.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def parse_args():
5959

6060
parser.add_argument('--save_dir', dest='save_dir',
6161
help='directory to save models', default="/srv/share/jyang375/models",
62-
nargs=argparse.REMAINDER)
62+
type=str)
6363
parser.add_argument('--nw', dest='num_workers',
6464
help='number of worker to load data',
6565
default=0, type=int)

0 commit comments

Comments
 (0)