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

fix(trainer): 修复回调函数合并逻辑

This commit is contained in:
ViperEkura 2026-01-08 21:56:44 +08:00
parent 4da70785b5
commit ff5c8a71f5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
khaosz/trainer/trainer.py
View File

@ -22,7 +22,8 @@ class Trainer:
callbacks: Optional[List[TrainCallback]] = None callbacks: Optional[List[TrainCallback]] = None
): ):
self.train_config = train_config self.train_config = train_config
self.callbacks = callbacks or self._get_default_callbacks() default_callbacks = self._get_default_callbacks()
self.callbacks = default_callbacks + callbacks if callbacks else default_callbacks
def _get_default_callbacks(self) -> List[TrainCallback]: def _get_default_callbacks(self) -> List[TrainCallback]:
train_config = self.train_config train_config = self.train_config