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

This commit is contained in:
Jonatas Grosman
2022-01-31 11:50:56 -03:00
committed by GitHub
parent 3254080d45
commit f624249d8b

View File

@@ -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,
)