!9286 add warmup_steps to maskrcnn config

From: @gengdongjie Reviewed-by: @linqingke,@oacjiewen Signed-off-by: @linqingke
2020-12-02 09:23:54 +08:00 · 2020-12-02 09:23:54 +08:00 · 7aae952ca3
parent e706cb2512 3deed63b56
commit 7aae952ca3
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/maskrcnn/src/config.py
+++ b/model_zoo/official/cv/maskrcnn/src/config.py
@ -121,7 +121,7 @@ config = ed({
    "base_lr": 0.02,
    "base_step": 58633,
    "total_epoch": 13,
-    "warmup_mode": "linear",
+    "warmup_step": 500,
    "warmup_ratio": 1/3.0,
    "sgd_momentum": 0.9,