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 38420f33f6fca1a32730a3b28a5208df0b113bae..ef82cb3263de1e36df32e00128388676da83f750 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 fe8a0cdf85ce840ff71dab2278a3c35cf54f7554..80b6b1c05888e5b5176744f7d15560531f405361 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(