Michael Diskin 4 年之前
父節點
當前提交
8d6ec69110
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      examples/albert/run_trainer.py

+ 2 - 0
examples/albert/run_trainer.py

@@ -82,6 +82,8 @@ def get_optimizer_and_scheduler(training_args, model):
         lr=training_args.learning_rate,
         eps=training_args.adam_epsilon,
         weight_decay=training_args.weight_decay,
+        relative_step=False,
+
     )
 
     scheduler = get_linear_schedule_with_warmup(