AstrAI/astrai/config
ViperEkura 2c2697390d feat: 新增 GradientCheckpointingCallback
- TrainConfig.gradient_checkpointing_modules 指定模块类型
- apply 递归遍历,兼容 DDP,不硬编码模型结构
- modules=None 时静默跳过,零开销
2026-05-17 18:21:05 +08:00
..
__init__.py refactor: Transformer更名为AutoRegressiveLM并新增EmbeddingEncoder 2026-05-17 15:29:20 +08:00
base.py refactor: Config序列化统一BaseConfig基类 2026-05-16 22:06:39 +08:00
model_config.py refactor: Transformer更名为AutoRegressiveLM并新增EmbeddingEncoder 2026-05-17 15:29:20 +08:00
train_config.py feat: 新增 GradientCheckpointingCallback 2026-05-17 18:21:05 +08:00