- return args.legacy_learning_rate * it / args.nb_warmup_iter
- elif it < args.nb_large_lr_epochs:
- return args.legacy_learning_rate
+ return args.legacy_large_lr * it / args.nb_warmup_iter
+ elif n_epoch < args.legacy_nb_epoch_large_lr:
+ return args.legacy_large_lr