From f3f304ee9c94b82e80713d73a9608af6b5e9e31b Mon Sep 17 00:00:00 2001 From: Chris Date: Tue, 12 Sep 2017 10:12:42 -0600 Subject: [PATCH] bug fix --- test_pretrained.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test_pretrained.py b/test_pretrained.py index eabc408..9c30121 100644 --- a/test_pretrained.py +++ b/test_pretrained.py @@ -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")