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

pt: add explicit decay_rate for lr #3445

Merged
merged 7 commits into from
Mar 20, 2024

Conversation

iProzd
Copy link
Collaborator

@iProzd iProzd commented Mar 11, 2024

This is for multitask training, when explicitly setting decay_rate is much more convenient for long training.

Copy link

codecov bot commented Mar 11, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 77.63%. Comparing base (2851fb9) to head (dd4a593).

Additional details and impacted files
@@            Coverage Diff             @@
##            devel    #3445      +/-   ##
==========================================
+ Coverage   77.49%   77.63%   +0.13%     
==========================================
  Files         432      399      -33     
  Lines       37164    32365    -4799     
  Branches     1620      909     -711     
==========================================
- Hits        28801    25127    -3674     
+ Misses       7495     6710     -785     
+ Partials      868      528     -340     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

deepmd/pt/utils/learning_rate.py Show resolved Hide resolved
deepmd/pt/utils/learning_rate.py Outdated Show resolved Hide resolved
deepmd/pt/utils/learning_rate.py Outdated Show resolved Hide resolved
deepmd/tf/utils/learning_rate.py Outdated Show resolved Hide resolved
deepmd/utils/argcheck.py Show resolved Hide resolved
@iProzd iProzd changed the title pt: add explicit decay_rate and min_lr for lr pt: add explicit decay_rate for lr Mar 13, 2024
source/tests/pt/test_lr.py Outdated Show resolved Hide resolved
@wanghan-iapcm wanghan-iapcm added this pull request to the merge queue Mar 20, 2024
Merged via the queue into deepmodeling:devel with commit 47366f6 Mar 20, 2024
48 checks passed
@iProzd iProzd deleted the add_lr_decay branch March 28, 2024 07:40
@njzjz njzjz mentioned this pull request Apr 2, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants