Unverified Commit f624249d authored by Jonatas Grosman's avatar Jonatas Grosman Committed by GitHub
Browse files

[Robust Speech Challenge] Add missing LR parameter (#15428)

parent 3254080d
......@@ -677,6 +677,7 @@ def main():
]
optimizer = bnb.optim.Adam8bit(
params=optimizer_grouped_parameters,
lr=training_args.learning_rate,
betas=(training_args.adam_beta1, training_args.adam_beta2),
eps=training_args.adam_epsilon,
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment