feat: 优化工厂模式的实现

2026-04-04 15:49:46 +08:00
parent aa5e03d7f6
commit 3346c75584
11 changed files with 228 additions and 142 deletions
@@ -1,12 +1,8 @@
 from astrai.trainer.schedule import BaseScheduler, SchedulerFactory
 from astrai.trainer.strategy import BaseStrategy, StrategyFactory
 from astrai.trainer.train_callback import (
-    CheckpointCallback,
-    GradientClippingCallback,
-    MetricLoggerCallback,
-    ProgressBarCallback,
-    SchedulerCallback,
    TrainCallback,
+    CallbackFactory,
 )
 from astrai.trainer.trainer import Trainer

@@ -19,11 +15,7 @@ __all__ = [
    # Scheduler factory
    "SchedulerFactory",
    "BaseScheduler",
-    # Callbacks
+    # Callback factory
    "TrainCallback",
-    "GradientClippingCallback",
-    "SchedulerCallback",
-    "CheckpointCallback",
-    "ProgressBarCallback",
-    "MetricLoggerCallback",
+    "CallbackFactory",
 ]