mixnet-s-classification.yaml

model:
  task: classification
  name: mixnet_s
  checkpoint:
    use_pretrained: True
    load_head: False
    path: ~
    fx_model_path: ~
    optimizer_path: ~
  freeze_backbone: False
  architecture:
    full: ~ # auto
    backbone:
      name: mixnet
      params:
        stem_channels: 16
        wid_mul: 1.0
        dep_mul: 1.0
        dropout_rate: 0.
      stage_params: 
        -
          expansion_ratio: [1, 6, 3]
          out_channels: [16, 24, 24]
          num_blocks: [1, 1, 1]
          kernel_sizes: [[3], [3], [3]]
          num_exp_groups: [1, 2, 2]
          num_poi_groups: [1, 2, 2]
          stride: [1, 2, 1]
          act_type: ["relu", "relu", "relu"]
          se_reduction_ratio: [~, ~, ~]
        -
          expansion_ratio: [6, 6]
          out_channels: [40, 40]
          num_blocks: [1, 3]
          kernel_sizes: [[3, 5, 7], [3, 5]]
          num_exp_groups: [1, 2]
          num_poi_groups: [1, 2]
          stride: [2, 1]
          act_type: ["swish", "swish"]
          se_reduction_ratio: [2, 2]
        -
          expansion_ratio: [6, 6, 6, 3]
          out_channels: [80, 80, 120, 120]
          num_blocks: [1, 2, 1, 2]
          kernel_sizes: [[3, 5, 7], [3, 5], [3, 5, 7], [3, 5, 7, 9]]
          num_exp_groups: [1, 1, 2, 2]
          num_poi_groups: [2, 2, 2, 2]
          stride: [2, 1, 1, 1]
          act_type: ["swish", "swish", "swish", "swish"]
          se_reduction_ratio: [4, 4, 2, 2]
        -
          expansion_ratio: [6, 6]
          out_channels: [200, 200]
          num_blocks: [1, 2]
          kernel_sizes: [[3, 5, 7, 9, 11], [3, 5, 7, 9]]
          num_exp_groups: [1, 1]
          num_poi_groups: [1, 2]
          stride: [2, 1]
          act_type: ["swish", "swish"]
          se_reduction_ratio: [2, 2]
    head:
      name: fc
      params:
        num_layers: 1
        intermediate_channels: ~
        act_type: ~
        dropout_prob: 0.
  losses:
    - criterion: cross_entropy
      label_smoothing: 0.1
      weight: ~