fix bug in inference

2023-07-21 10:53:29 +08:00
parent 0da6d1214a
commit 471dd5831f
1 changed files with 1 additions and 1 deletions
@@ -21,7 +21,7 @@ def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--num_gpus', type=int, dest='num_gpus')
    parser.add_argument('--exp_name', type=str, default='output/test')
-    parser.add_argument('--pretrained_model', type=int, default=0)
+    parser.add_argument('--pretrained_model', type=str, default=0)
    parser.add_argument('--testset', type=str, default='EHF')
    parser.add_argument('--agora_benchmark', type=str, default='na')
    parser.add_argument('--img_path', type=str, default='input.png')