Parcourir la source

bug fix

master
Chris il y a 8 ans
Parent
révision
f3f304ee9c
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. +1
    -1
      test_pretrained.py

+ 1
- 1
test_pretrained.py Voir le fichier

@@ -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")


Chargement…
Annuler
Enregistrer