yjh0410 vor 1 Jahr
Ursprung
Commit
e84da16895
1 geänderte Dateien mit 8 neuen und 4 gelöschten Zeilen
  1. 8 4
      odlab/config/__init__.py

+ 8 - 4
odlab/config/__init__.py

@@ -6,13 +6,17 @@ from .detr_config      import build_detr_config
 def build_config(args):
     # FCOS
     if "fcos" in args.model:
-        return build_fcos_config(args)
+        cfg = build_fcos_config(args)
     # YOLOF
     elif "yolof" in args.model:
-        return build_yolof_config(args)
+        cfg = build_yolof_config(args)
     # DETR
     elif "detr" in args.model:
-        return build_detr_config(args)
-    
+        cfg = build_detr_config(args)
     else:
         raise NotImplementedError('Unknown Model: {}'.format(args.model))
+
+    # Print model config
+    cfg.print_config()
+
+    return cfg