chore: 修改类名，优化导入顺序

2026-04-05 22:27:57 +08:00
parent 9b22b1651e
commit 39766aa1dc
21 changed files with 40 additions and 47 deletions
@@ -2,10 +2,10 @@ import argparse
 import os
 from functools import partial

+import safetensors.torch as st
 import torch
 import torch.nn as nn
 import torch.optim as optim
-import safetensors.torch as st
 from torch.nn.parallel import DistributedDataParallel as DDP

 from astrai.config import ModelConfig, TrainConfig