mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-22 22:02:51 +08:00
Merge pull request #4006 from Uminosachi/scheduler-kwargs
Set scheduler_specific_kwargs to get_scheduler Former-commit-id: 0f01500b68396533ddb00dc6e92e6e26be0ce9e5
This commit is contained in:
commit
f6f1c4eacb
@ -379,6 +379,7 @@ def create_custom_scheduler(
|
||||
optimizer=optimizer_dict[param],
|
||||
num_warmup_steps=training_args.get_warmup_steps(num_training_steps),
|
||||
num_training_steps=num_training_steps,
|
||||
scheduler_specific_kwargs=training_args.lr_scheduler_kwargs,
|
||||
)
|
||||
|
||||
def scheduler_hook(param: "torch.nn.Parameter"):
|
||||
|
Loading…
x
Reference in New Issue
Block a user