refactor: 优化参数传递，清理导入样式

2026-04-03 22:06:32 +08:00
parent 3a7d98a950
commit 0852b852f8
51 changed files with 299 additions and 434 deletions
@@ -1,14 +1,7 @@
 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",
@@ -16,9 +9,4 @@ __all__ = [
    # Model configuration
    "ModelConfig",
    "TrainConfig",
-    # Schedule configuration
-    "ScheduleConfig",
-    "CosineScheduleConfig",
-    "SGDRScheduleConfig",
-    "ScheduleConfigFactory",
 ]