chore: 更新项目名称

2026-03-31 09:34:11 +08:00
parent 780b9e1855
commit 2e009cf59a
51 changed files with 118 additions and 118 deletions
@@ -0,0 +1,30 @@
+from astrai.trainer.trainer import Trainer
+from astrai.trainer.strategy import StrategyFactory, BaseStrategy
+from astrai.trainer.schedule import SchedulerFactory, BaseScheduler
+
+from astrai.trainer.train_callback import (
+    TrainCallback,
+    GradientClippingCallback,
+    SchedulerCallback,
+    CheckpointCallback,
+    ProgressBarCallback,
+    MetricLoggerCallback,
+)
+
+__all__ = [
+    # Main trainer
+    "Trainer",
+    # Strategy factory
+    "StrategyFactory",
+    "BaseStrategy",
+    # Scheduler factory
+    "SchedulerFactory",
+    "BaseScheduler",
+    # Callbacks
+    "TrainCallback",
+    "GradientClippingCallback",
+    "SchedulerCallback",
+    "CheckpointCallback",
+    "ProgressBarCallback",
+    "MetricLoggerCallback",
+]