Skip to content

Commit d002764

Browse files
author
Drasick
committed
1. fix trainer.py 的错误,补充 self._config = trainer_cfg 防止报错
2. gitignore 加上 ckpt
1 parent a50567a commit d002764

2 files changed

Lines changed: 4 additions & 0 deletions

File tree

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,3 +136,5 @@ GroupedGEMM/
136136

137137
xtuner_train/
138138

139+
140+
ckpt

xtuner/v1/train/trainer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,8 @@ def __init__(
481481
hooks_config: HooksConfig = HooksConfig(),
482482
internal_metrics_cfg: InternalMetricsConfig | None = None,
483483
):
484+
self._config = trainer_cfg
485+
484486
self._do_clip = do_clip
485487
self._grad_norm_dtype = grad_norm_dtype
486488
self._dataloader_config = dataloader_cfg

0 commit comments

Comments
 (0)