yjh0410 1 年間 前
コミット
e84da16895
1 ファイル変更8 行追加4 行削除
  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