From 3deed63b56a9170115813d6098beba7dcee9993b Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Tue, 1 Dec 2020 14:43:17 +0800 Subject: [PATCH] revert warpup_steps in maskrcnn config that deleted by mistake --- model_zoo/official/cv/maskrcnn/src/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/maskrcnn/src/config.py b/model_zoo/official/cv/maskrcnn/src/config.py index bde4c302705..f4dccaa56be 100644 --- 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,