Internal change

PiperOrigin-RevId: 440544815

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_fpn_600epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: fpn
+    detection_head:
+      num_fcs: 2
+    norm_activation:
+      activation: swish
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [1062734, 1090458]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 1108940

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_fpn_72epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: fpn
+    detection_head:
+      num_fcs: 2
+    norm_activation:
+      activation: swish
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [88704, 125664]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 133056

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_fpn_cascade_600epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: fpn
+    detection_head:
+      cascade_class_ensemble: true
+      class_agnostic_bbox_pred: true
+      num_fcs: 2
+    input_size: [1280, 1280, 3]
+    norm_activation:
+      activation: swish
+    roi_sampler:
+      cascade_iou_thresholds: [0.7, 0.8]
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [1062734, 1090458]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 1108940

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_fpn_cascade_72epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: fpn
+    detection_head:
+      cascade_class_ensemble: true
+      class_agnostic_bbox_pred: true
+      num_fcs: 2
+    input_size: [1280, 1280, 3]
+    norm_activation:
+      activation: swish
+    roi_sampler:
+      cascade_iou_thresholds: [0.7, 0.8]
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [88704, 125664]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 133056

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_nasfpn_600epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: nasfpn
+    detection_head:
+      num_fcs: 2
+    include_mask: false
+    max_level: 7
+    min_level: 3
+    norm_activation:
+      activation: swish
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [1062734, 1090458]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 1108940

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_nasfpn_72epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: nasfpn
+    detection_head:
+      num_fcs: 2
+    include_mask: false
+    max_level: 7
+    min_level: 3
+    norm_activation:
+      activation: swish
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [88704, 125664]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 133056

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_nasfpn_cascade_600epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: nasfpn
+    detection_head:
+      cascade_class_ensemble: true
+      class_agnostic_bbox_pred: true
+      num_fcs: 2
+    input_size: [1280, 1280, 3]
+    max_level: 7
+    min_level: 3
+    norm_activation:
+      activation: swish
+    roi_sampler:
+      cascade_iou_thresholds: [0.7, 0.8]
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [1062734, 1090458]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 1108940

official/projects/backbone_reuse/configs/experiments/faster_rcnn/fastrcnn_resnet101_nasfpn_cascade_72epochs.yaml

+task:
+  # init_checkpoint: 'a_pretrained_backbone_checkpoint'
+  init_checkpoint_modules: backbone
+  freeze_backbone: true
+  model:
+    backbone:
+      resnet:
+        model_id: 101
+        replace_stem_max_pool: true
+        resnetd_shortcut: true
+        scale_stem: true
+        se_ratio: 0.25
+        stem_type: v1
+      type: resnet
+    decoder:
+      type: nasfpn
+    detection_head:
+      cascade_class_ensemble: true
+      class_agnostic_bbox_pred: true
+      num_fcs: 2
+    input_size: [1280, 1280, 3]
+    max_level: 7
+    min_level: 3
+    norm_activation:
+      activation: swish
+    roi_sampler:
+      cascade_iou_thresholds: [0.7, 0.8]
+  train_data:
+    global_batch_size: 64
+    parser:
+      aug_rand_hflip: true
+      aug_scale_max: 2.0
+      aug_scale_min: 0.1
+trainer:
+  optimizer_config:
+    learning_rate:
+      stepwise:
+        boundaries: [88704, 125664]
+        name: PiecewiseConstantDecay
+        offset: 0
+        values: [0.16, 0.016, 0.0016]
+      type: stepwise
+  steps_per_loop: 1848
+  summary_interval: 1848
+  train_steps: 133056