fix args in eval_script.py

leondgarse · Apr 13, 2022 · 7943272 · 7943272
1 parent b8699b6
commit 7943272
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eval_script.py b/eval_script.py
@@ -32,6 +32,7 @@ def parse_arguments(argv):
         help="Pretrianed weights if not from h5. Could be [imagenet, noisy_student, imagenet21k, imagenet21k-ft1k, imagenet_sam], None for model.pretrained",
     )
 
+    args = parser.parse_known_args(argv)[0]
     return args