加了--debug_input参数

2026-02-20 14:59:44 +08:00
parent d85cce8a52
commit 88d0cc5ca2
2 changed files with 31 additions and 5 deletions
--- a/detr/main.py
+++ b/detr/main.py
@@ -80,7 +80,7 @@ def get_args_parser():

 def build_ACT_model_and_optimizer(args_override):
    parser = argparse.ArgumentParser('DETR training and evaluation script', parents=[get_args_parser()])
-    args = parser.parse_args()
+    args, _ = parser.parse_known_args()

    for k, v in args_override.items():
        setattr(args, k, v)
@@ -103,7 +103,7 @@ def build_ACT_model_and_optimizer(args_override):

 def build_CNNMLP_model_and_optimizer(args_override):
    parser = argparse.ArgumentParser('DETR training and evaluation script', parents=[get_args_parser()])
-    args = parser.parse_args()
+    args, _ = parser.parse_known_args()

    for k, v in args_override.items():
        setattr(args, k, v)