set default O2 for profile

Author: Mark-ZhouWX

research/segment-anything/configs/profile_coco_box_finetune.yaml

@@ -5,7 +5,7 @@ device: Ascend
 mode: 0  # 0: graph, 1: pynative
 work_root: &work_root ./work_dir/
 log_level: info
-
+amp_level: O2
 
 # ---------------------------------------------
 # Part2: module setting

research/segment-anything/configs/profile_flare_box_finetune.yaml

@@ -5,7 +5,7 @@ device: Ascend
 mode: 0  # 0: graph, 1: pynative
 work_root: &work_root ./work_dir/
 log_level: info
-
+amp_level: O2
 
 # ---------------------------------------------
 # Part2: module setting

research/segment-anything/segment_anything/automatic_mask_generator.py

@@ -7,7 +7,6 @@
 import numpy as np
 import mindspore as ms
 from mindspore import ops
-from torchvision.ops.boxes import batched_nms  # type: ignore
 
 from typing import Any, Dict, List, Optional, Tuple
 

research/segment-anything/segment_anything/utils/callbacks.py

@@ -176,7 +176,7 @@ def on_train_epoch_end(self, run_context: RunContext):
 
 @CALLBACK_REGISTRY.registry_module()
 class Profiler(Callback):
-    def __init__(self, start_step=1, end_step=2, exit_after_analyze=True, out_dir='./data'):
+    def __init__(self, start_step=1, end_step=2, exit_after_analyze=True, out_dir='./my_prof'):
         self.start_step = start_step
         self.end_step = end_step
         self.exit_after_analyze = exit_after_analyze