From b9d80a477644599c0495e4c2b3f29b3d1a00be9d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E4=BC=9F=E6=A0=B9?= <1101204667@qq.com>
Date: Fri, 15 Apr 2022 08:56:10 +0000
Subject: [PATCH] update defaults.py.

---
 .../MaskRCNN_ID0101_for_PyTorch/detectron2/engine/defaults.py  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/engine/defaults.py b/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/engine/defaults.py
index c3a6934929..6914d8dabc 100644
--- a/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/engine/defaults.py
+++ b/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/engine/defaults.py
@@ -317,7 +317,8 @@ class DefaultTrainer(SimpleTrainer):
                 model,
                 optimizer,
                 opt_level=cfg.OPT_LEVEL,
-                loss_scale=cfg.LOSS_SCALE_VALUE)
+                loss_scale=cfg.LOSS_SCALE_VALUE,
+                combine_grad=True)
 
         data_loader = self.build_train_loader(cfg)
 
-- 
Gitee