From f624249d8b49c50e507077d801062c7fb58797e9 Mon Sep 17 00:00:00 2001 From: Jonatas Grosman Date: Mon, 31 Jan 2022 11:50:56 -0300 Subject: [PATCH] [Robust Speech Challenge] Add missing LR parameter (#15428) --- .../robust-speech-event/run_speech_recognition_ctc_bnb.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/research_projects/robust-speech-event/run_speech_recognition_ctc_bnb.py b/examples/research_projects/robust-speech-event/run_speech_recognition_ctc_bnb.py index 20f9ba0c5b..2317367e7c 100755 --- a/examples/research_projects/robust-speech-event/run_speech_recognition_ctc_bnb.py +++ b/examples/research_projects/robust-speech-event/run_speech_recognition_ctc_bnb.py @@ -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, )