chore: 将data 模块命名为dataset

2026-04-04 16:16:27 +08:00
parent bd9741dc5f
commit 9c31d78a22
8 changed files with 7 additions and 7 deletions
@@ -8,7 +8,7 @@ import torch.optim as optim
 from torch.nn.parallel import DistributedDataParallel as DDP

 from astrai.config import ModelParameter, TrainConfig
-from astrai.data import DatasetFactory
+from astrai.dataset import DatasetFactory
 from astrai.parallel import get_rank
 from astrai.trainer import SchedulerFactory, Trainer