disable learning rate warmup

5d7d327e · srihari-humbarwadi · 5893d611 · 5d7d327e
Unverified Commit 5d7d327e authored Mar 20, 2022 by srihari-humbarwadi
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py ...ta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py +1 -1

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
@@ -297,7 +297,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
              'warmup': {
                  'type': 'linear',
                  'linear': {
-                      'warmup_steps': 4 * steps_per_epoch,
+                      'warmup_steps': 0,
                      'warmup_learning_rate': 0
                  }
              }