Selaa lähdekoodia

bug fix

master
Chris 8 vuotta sitten
vanhempi
commit
f3f304ee9c
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. +1
    -1
      test_pretrained.py

+ 1
- 1
test_pretrained.py Näytä tiedosto

@@ -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…
Peruuta
Tallenna