Division to num_train_optimizer of global_step in lr_this_step is removed.

This commit is contained in:
burcturkoglu committed 2019-05-09 10:57:03 +03:00
1 parent fa37b4da77
commit 00c7fd2b79
5 files changed
+5 -10

No files matched your search

+1 -2
View File
@@ -1016,8 +1016,7 @@ def main():
if args.fp16:
# modify learning rate with special warm up BERT uses
# if args.fp16 is False, BertAdam is used and handles this automatically
lr_this_step = args.learning_rate * warmup_linear.get_lr(global_step,
args.warmup_proportion)
lr_this_step = args.learning_rate * warmup_linear.get_lr(global_step, args.warmup_proportion)
for param_group in optimizer.param_groups:
param_group['lr'] = lr_this_step
optimizer.step()