From 180ac93c207220763df910731da771be2435a891 Mon Sep 17 00:00:00 2001 From: sunyinlei Date: Wed, 21 Dec 2022 09:37:10 +0800 Subject: [PATCH] =?UTF-8?q?bugfix:=E4=BF=AE=E5=A4=8DMaskRCNN=E5=92=8CCasca?= =?UTF-8?q?dedMaskRCNN=E5=9C=A8B020=E7=89=88=E6=9C=AC=E4=BA=8C=E8=BF=9B?= =?UTF-8?q?=E5=88=B6=E4=BD=BF=E8=83=BD=E4=B8=8B=E5=8A=9F=E8=83=BD=E6=8A=A5?= =?UTF-8?q?=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../cv/detection/Faster_Mask_RCNN_for_PyTorch/tools/train_net.py | 1 + PyTorch/contrib/cv/detection/CascadedMaskRCNN/tools/train_net.py | 1 + 2 files changed, 2 insertions(+) diff --git a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/tools/train_net.py b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/tools/train_net.py index 38420f33f6..ef82cb3263 100755 --- a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/tools/train_net.py +++ b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/tools/train_net.py @@ -186,6 +186,7 @@ def main(args): if __name__ == "__main__": + torch.npu.set_compile_mode(jit_compile=True) args = default_argument_parser().parse_args() print("Command Line Args:", args) launch( diff --git a/PyTorch/contrib/cv/detection/CascadedMaskRCNN/tools/train_net.py b/PyTorch/contrib/cv/detection/CascadedMaskRCNN/tools/train_net.py index fe8a0cdf85..80b6b1c058 100644 --- a/PyTorch/contrib/cv/detection/CascadedMaskRCNN/tools/train_net.py +++ b/PyTorch/contrib/cv/detection/CascadedMaskRCNN/tools/train_net.py @@ -182,6 +182,7 @@ def main(args): if __name__ == "__main__": + torch.npu.set_compile_mode(jit_compile=True) args = default_argument_parser().parse_args() print("Command Line Args:", args) launch( -- Gitee