fix bug of callback hack update

Author: Mark-ZhouWX

research/segment-anything/configs/profile_coco_box_finetune.yaml

@@ -0,0 +1,71 @@
+#---------------------------------------------
+# Part 1: system basic config setting
+distributed: False
+device: Ascend
+mode: 0  # 0: graph, 1: pynative
+work_root: &work_root ./work_dir/
+log_level: info
+
+
+# ---------------------------------------------
+# Part2: module setting
+loss_manager:
+#  type: fixed  # dynamic or
+#  scale_sense: 1024
+  loss_scaler:
+    type: dynamic
+  grad_clip: False
+
+
+optimizer:
+  type: segment_anything.optim.optimizer.AdamW
+  weight_decay: 1e-4
+  group_param:
+
+  lr_scheduler:
+    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    learning_rate: 8e-6
+    warmup_steps: 250
+    decay_steps: [ 60000, 86666 ]
+    decay_factor: 10
+
+
+network:
+  model:
+    type: vit_b
+    checkpoint: ./models/sam_vit_b-35e4849c.ckpt
+    freeze:
+      image_encoder: True
+      prompt_encoder: True
+
+  loss:
+    type: segment_anything.modeling.loss.SAMLoss
+
+
+train_loader:
+  dataset:
+    type: segment_anything.dataset.dataset.COCODataset
+    data_dir: ./datasets/coco2017/train2017
+    annotation_path: ./datasets/coco2017/annotations/instances_train2017.json
+    transform_pipeline:
+      - type: segment_anything.dataset.transform.ImageResizeAndPad
+        target_size: 1024
+      - type: segment_anything.dataset.transform.ImageNorm
+        hwc2chw: True
+      - type: segment_anything.dataset.transform.LabelPad
+        gt_size: 20
+    output_column: ['image', 'masks', 'boxes', 'valid_boxes']
+
+  model_column: ['image', 'boxes']  # columns for model cell input
+  loss_column:  ['masks', 'valid_boxes']  # columns for loss function input
+
+  shuffle: True
+  batch_size: 1
+  epoch_size: 50
+  drop_remainder: True
+  num_workers: 2
+  max_rowsize: 24  # 24M space for dataloader
+
+
+callback:
+  - type: segment_anything.utils.callbacks.Profiler

research/segment-anything/configs/profile_flare_box_finetune.yaml

@@ -0,0 +1,68 @@
+#---------------------------------------------
+# Part 1: system basic config setting
+distributed: False
+device: Ascend
+mode: 0  # 0: graph, 1: pynative
+work_root: &work_root ./work_dir/
+log_level: info
+
+
+# ---------------------------------------------
+# Part2: module setting
+loss_manager:
+#  type: fixed  # dynamic or
+#  scale_sense: 1024
+  loss_scaler:
+    type: dynamic
+  grad_clip: False
+
+
+optimizer:
+  type: segment_anything.optim.optimizer.AdamW
+  weight_decay: 1e-4
+  group_param:
+
+  lr_scheduler:
+    type: segment_anything.optim.scheduler.SAMDynamicDecayLR
+    learning_rate: 8e-6
+    warmup_steps: 250
+    decay_steps: [ 60000, 86666 ]
+    decay_factor: 10
+
+
+network:
+  model:
+    type: vit_b
+    checkpoint: ./models/sam_vit_b-35e4849c.ckpt
+    freeze:
+      image_encoder: True
+      prompt_encoder: True
+
+  loss:
+    type: segment_anything.modeling.loss.SAMLoss
+
+
+train_loader:
+  dataset:
+    type: segment_anything.dataset.dataset.FLAREDataset
+    data_dir: ./datasets/FLARE22Train_processed/train/
+    transform_pipeline:
+      - type: segment_anything.dataset.transform.BinaryMaskFromInstanceSeg
+      - type: segment_anything.dataset.transform.BoxFormMask
+      - type: segment_anything.dataset.transform.LabelPad
+        gt_size: 20
+    output_column: ['image', 'masks', 'boxes', 'valid_boxes' ]
+
+  model_column: ['image', 'boxes' ]  # columns for model cell input
+  loss_column:  ['masks', 'valid_boxes' ]  # columns for loss function input
+
+  shuffle: True
+  batch_size: 1
+  epoch_size: 50
+  drop_remainder: True
+  num_workers: 2
+  max_rowsize: 64  # 24M space for dataloader
+
+
+callback:
+  - type: segment_anything.utils.callbacks.Profiler

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

@@ -105,7 +105,7 @@ def update_rank_to_dataloader_config(rank_id, rank_size, args_train_loader, args
     # a hack implementation to update runtime-defined setting in callback args
     if arg_callback is not None:
         for cb in arg_callback:
-            if cb.type == 'EvalWhileTrain':
+            if cb.type.endswith('EvalWhileTrain'):
                 cb.data_loader = args_eval_loader
 
 
@@ -130,7 +130,7 @@ def set_directory_and_log(main_device, rank_id, rank_size, work_root, log_level,
     # a hack implementation to update runtime-defined setting in callback args
     if args_callback is not None:
         for cb in args_callback:
-            if cb.type == 'SaveCkpt':
+            if cb.type.endswith('SaveCkpt'):
                 hack_list = {'save_dir': save_dir, 'main_device': main_device}
                 cb.update(hack_list)
     return save_dir