birth

classification/configs/flash_intern_image_b_1k_224.yaml

@@ -0,0 +1,20 @@
+DATA:
+  IMG_ON_MEMORY: False
+MODEL:
+  TYPE: flash_intern_image
+  DROP_PATH_RATE: 0.5
+  FLASH_INTERN_IMAGE:
+    CORE_OP: 'DCNv4'
+    DEPTHS: [4, 4, 21, 4]
+    GROUPS: [7, 14, 28, 56]
+    CHANNELS: 112
+    LAYER_SCALE: 1e-5
+    OFFSET_SCALE: 0.5
+    MLP_RATIO: 4.0
+    POST_NORM: True
+    DW_KERNEL_SIZE: 3
+TRAIN:
+  EMA:
+    ENABLE: True
+    DECAY: 0.9999
+  BASE_LR: 5e-4

classification/configs/flash_intern_image_l_22kto1k_384.yaml

@@ -0,0 +1,40 @@
+DATA:
+  IMG_SIZE: 384
+  IMG_ON_MEMORY: False
+AUG:
+  MIXUP: 0.0
+  CUTMIX: 0.0
+  REPROB: 0.0
+MODEL:
+  TYPE: flash_intern_image
+  DROP_PATH_RATE: 0.1
+  LABEL_SMOOTHING: 0.3
+  FLASH_INTERN_IMAGE:
+    CORE_OP: 'DCNv4'
+    DEPTHS: [5, 5, 22, 5]
+    GROUPS: [10, 20, 40, 80]
+    CHANNELS: 160
+    LAYER_SCALE: 1e-5
+    OFFSET_SCALE: 2.0
+    MLP_RATIO: 4.0
+    POST_NORM: True
+    DW_KERNEL_SIZE: 3
+    DCN_OUTPUT_BIAS: True
+    MLP_FC2_BIAS: True
+TRAIN:
+  EMA:
+    ENABLE: true
+    DECAY: 0.9999
+  EPOCHS: 20
+  WARMUP_EPOCHS: 2
+  WEIGHT_DECAY: 0.05
+  BASE_LR: 2e-05 # 512
+  WARMUP_LR: .0
+  MIN_LR: .0
+  LR_LAYER_DECAY: true
+  LR_LAYER_DECAY_RATIO: 0.9
+  USE_CHECKPOINT: true
+  OPTIMIZER:
+    DCN_LR_MUL: 0.1
+AMP_OPT_LEVEL: O0
+EVAL_FREQ: 1

classification/configs/flash_intern_image_s_1k_224.yaml

@@ -0,0 +1,20 @@
+DATA:
+  IMG_ON_MEMORY: False
+MODEL:
+  TYPE: flash_intern_image
+  DROP_PATH_RATE: 0.4
+  FLASH_INTERN_IMAGE:
+    CORE_OP: 'DCNv4'
+    DEPTHS: [4, 4, 21, 4]
+    GROUPS: [5, 10, 20, 40]
+    CHANNELS: 80
+    LAYER_SCALE: 1e-5
+    OFFSET_SCALE: 1.0
+    MLP_RATIO: 4.0
+    POST_NORM: True
+    DW_KERNEL_SIZE: 3
+TRAIN:
+  EMA:
+    ENABLE: True
+    DECAY: 0.9999
+  BASE_LR: 5e-4

classification/configs/flash_intern_image_t_1k_224.yaml

@@ -0,0 +1,17 @@
+DATA:
+  IMG_ON_MEMORY: False
+MODEL:
+  TYPE: flash_intern_image
+  DROP_PATH_RATE: 0.1
+  FLASH_INTERN_IMAGE:
+    CORE_OP: 'DCNv4'
+    DEPTHS: [4, 4, 18, 4]
+    GROUPS: [4, 8, 16, 32]
+    CHANNELS: 64
+    OFFSET_SCALE: 1.0
+    MLP_RATIO: 4.0
+TRAIN:
+  EMA:
+    ENABLE: True
+    DECAY: 0.9999
+  BASE_LR: 5e-4