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,24 @@
+from astrai.config.model_config import ModelConfig
+from astrai.config.param_config import BaseModelIO, ModelParameter
+from astrai.config.schedule_config import (
+    ScheduleConfig,
+    CosineScheduleConfig,
+    SGDRScheduleConfig,
+    ScheduleConfigFactory,
+)
+from astrai.config.train_config import TrainConfig
+
+
+__all__ = [
+    # Base I/O
+    "BaseModelIO",
+    "ModelParameter",
+    # Model configuration
+    "ModelConfig",
+    "TrainConfig",
+    # Schedule configuration
+    "ScheduleConfig",
+    "CosineScheduleConfig",
+    "SGDRScheduleConfig",
+    "ScheduleConfigFactory",
+]