diff --git a/official/cv/maskrcnn/default_config.yaml b/official/cv/maskrcnn/default_config.yaml index b87bf30d31f79659223ec53654125a50f88ac213..fd2c49c5433e60f6e660eab1bb9ba66131ecc699 100644 --- a/official/cv/maskrcnn/default_config.yaml +++ b/official/cv/maskrcnn/default_config.yaml @@ -137,7 +137,7 @@ sgd_momentum: 0.9 # train batch_size: 2 -loss_scale: 1 +loss_scale: 1024 momentum: 0.91 weight_decay: 0.0001 # 1e-4 pretrain_epoch_size: 0