Browse Source

bug fix

master
Chris 8 years ago
parent
commit
f3f304ee9c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test_pretrained.py

+ 1
- 1
test_pretrained.py View File

@@ -80,7 +80,7 @@ def get_args():
parser.add_argument("out_file", type=str,
help="Output file")

parser.add_argument("--out-dir", type=str, default=out,
parser.add_argument("--out-dir", type=str, default='out',
help="")
parser.add_argument("--gpu", type=int, default=0,
help="GPU to use for running the network")


Loading…
Cancel
Save