mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-11-07 03:12:13 +08:00
[train] fix denominator of ga in ksft loss (#9409)
This commit is contained in:
parent
8edd2622ce
commit
bd30c0003b
@ -83,6 +83,7 @@ def run_sft(
|
|||||||
**dataset_module,
|
**dataset_module,
|
||||||
**metric_module,
|
**metric_module,
|
||||||
)
|
)
|
||||||
|
trainer.model_accepts_loss_kwargs = False
|
||||||
|
|
||||||
# Training
|
# Training
|
||||||
if training_args.do_train:
|
if training_args.do_train:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user