NbAiLab
/

roberta_des_ada_128

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

pere commited on Dec 23, 2021

Commit

262ed92

•

1 Parent(s): 977847b

fix

Files changed (1) hide show

run_mlm_flax.py +1 -1

run_mlm_flax.py CHANGED Viewed

@@ -508,7 +508,7 @@ if __name__ == "__main__":
         init_value=0.0, end_value=training_args.learning_rate, transition_steps=training_args.warmup_steps
     )
-    if data_argsdata_argtatic_learning_rate:
         end_lr_value = training_args.learning_rate
     else:
         end_lr_value = 0

         init_value=0.0, end_value=training_args.learning_rate, transition_steps=training_args.warmup_steps
     )
+    if data_args.static_learning_rate:
         end_lr_value = training_args.learning_rate
     else:
         end_lr_value = 0