conv_cvnets.yaml

taskname: '+ MobileViTv2-2.0 plant'
common:
  enable_coreml_compatible_module: true
  results_loc: "conv"
  run_label: "finetune_mobilevit_plants"
  mixed_precision: true
  conversion:
    input_image_path: "/home/nme/WORK/plants_and_shrooms/plants_images/6541/a177a8d425913fac10075c76bc8332a0ad83d04eDsc_0063.jpg"
dataset:
  category: "classification"
model:
  classification:
    freeze_batch_norm: true
    finetune_pretrained_model: false
    pretrained: "results/finetune_mobilevit_plants/checkpoint_best.pt"
    n_classes: 3343
    name: "mobilevit_v2"
    mitv2:
      width_multiplier: 2.0
      attn_norm_layer: "layer_norm_2d"
    activation:
      name: "swish"
  normalization:
    name: "batch_norm"
    momentum: 0.1
  activation:
    name: "swish"
  layer:
    global_pool: "mean"
    conv_init:  "kaiming_normal"
    linear_init: "trunc_normal"
    linear_init_std_dev: 0.02

Convert cvnets to coreml