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(