Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

exp_lr_scheduler is missing #1

Open
gudovskiy opened this issue Nov 29, 2018 · 1 comment
Open

exp_lr_scheduler is missing #1

gudovskiy opened this issue Nov 29, 2018 · 1 comment

Comments

@gudovskiy
Copy link

Seems that you need this to utils.py:

def exp_lr_scheduler(optimizer, decay_ratio):
"""Decay learning rate by a factor of lr_decay every lr_decay_epoch epochs"""
for param_group in optimizer.param_groups:
param_group['lr'] *= decay_ratio
return optimizer

@gudovskiy
Copy link
Author

OK, I found this function in ./optimizer . Then, call should be:
hf_optm_utils.exp_lr_scheduler(optimizer, decay_ratio=args.lr_decay)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant