From 794327211ec59fc2174c4c177be37af01db78ba9 Mon Sep 17 00:00:00 2001 From: leondgarse Date: Wed, 13 Apr 2022 09:26:31 +0800 Subject: [PATCH] fix args in eval_script.py --- eval_script.py | 1 + 1 file changed, 1 insertion(+) diff --git a/eval_script.py b/eval_script.py index 85aba4e3..31df5525 100644 --- 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