chore: 更新项目名称

2026-03-31 09:34:11 +08:00
parent 780b9e1855
commit 2e009cf59a
51 changed files with 118 additions and 118 deletions
@@ -4,8 +4,8 @@ import torch.distributed as dist

 from torch.optim import AdamW
 from torch.optim.lr_scheduler import CosineAnnealingLR
-from khaosz.data.serialization import Checkpoint
-from khaosz.parallel.setup import get_rank, spawn_parallel_fn
+from astrai.data.serialization import Checkpoint
+from astrai.parallel.setup import get_rank, spawn_parallel_fn


 def test_single_process():