Skip to content

Commit 9725c7f

Browse files
Xiaofang Wangfacebook-github-bot
authored andcommitted
Add cfg.SOLVER.NUM_DECAYS in the default config
Summary: Pull Request resolved: #4535 cfg.SOLVER.NUM_DECAYS is needed for the WarmupStepWithFixedGammaLR schedule. Reviewed By: wat3rBro Differential Revision: D39363550 fbshipit-source-id: b2844d4c84b8538f59a958984627b9463b6d6324
1 parent 717ab9f commit 9725c7f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

detectron2/config/defaults.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -541,6 +541,8 @@
541541
_C.SOLVER.GAMMA = 0.1
542542
# The iteration number to decrease learning rate by GAMMA.
543543
_C.SOLVER.STEPS = (30000,)
544+
# Number of decays in WarmupStepWithFixedGammaLR schedule
545+
_C.SOLVER.NUM_DECAYS = 3
544546

545547
_C.SOLVER.WARMUP_FACTOR = 1.0 / 1000
546548
_C.SOLVER.WARMUP_ITERS = 1000

0 commit comments

Comments
 (0)