ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(callback): 解决循环导入问题

This commit is contained in:
ViperEkura 2025-09-29 13:08:41 +08:00
parent b2f3fefa1b
commit 5163d3a47a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
khaosz/trainer/callback.py
View File

@ -1,8 +1,10 @@
from tqdm import tqdm from tqdm import tqdm
from khaosz.core.parameter import Checkpoint from khaosz.core.parameter import Checkpoint
from khaosz.trainer.trainer import Trainer
from torch.nn.utils import clip_grad_norm_ from torch.nn.utils import clip_grad_norm_
from typing import cast from typing import cast, TYPE_CHECKING
if TYPE_CHECKING:
from khaosz.trainer.trainer import Trainer
class TrainerCallback: class TrainerCallback: