[Robust Speech Challenge] Add missing LR parameter (#15428)
This commit is contained in:
@@ -677,6 +677,7 @@ def main():
|
|||||||
]
|
]
|
||||||
optimizer = bnb.optim.Adam8bit(
|
optimizer = bnb.optim.Adam8bit(
|
||||||
params=optimizer_grouped_parameters,
|
params=optimizer_grouped_parameters,
|
||||||
|
lr=training_args.learning_rate,
|
||||||
betas=(training_args.adam_beta1, training_args.adam_beta2),
|
betas=(training_args.adam_beta1, training_args.adam_beta2),
|
||||||
eps=training_args.adam_epsilon,
|
eps=training_args.adam_epsilon,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user