fix: 修复路径问题

2026-03-20 21:14:02 +08:00
parent 361cdeb296
commit 3e8f2eba81
4 changed files with 4 additions and 4 deletions
@@ -4,7 +4,7 @@ import torch.distributed as dist

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

 def test_single_process():