feat: metric 参数通过 TrainConfig 传递

- TrainConfig 新增 log_dir/log_interval/metrics 配置字段 - metric_logger 调用改用 **kwargs 传递，BaseFactory.create 自动过滤
2026-05-19 17:50:24 +08:00
parent e0a3337c22
commit 45479b5731
2 changed files with 21 additions and 2 deletions
@@ -36,8 +36,14 @@ class Trainer:
                cfg.ckpt_interval,
                state_dict_fn=cfg.state_dict_fn,
            ),
+            CallbackFactory.create(
+                "metric_logger",
+                log_dir=cfg.log_dir,
+                save_interval=cfg.ckpt_interval,
+                log_interval=cfg.log_interval,
+                metrics=cfg.metrics,
+            ),
            CallbackFactory.create("progress_bar", cfg.n_epoch),
-            CallbackFactory.create("metric_logger", cfg.ckpt_dir, cfg.ckpt_interval),
            CallbackFactory.create("gradient_clipping", cfg.max_grad_norm),
            CallbackFactory.create("validation"),
        ]