cli-automl-image-classification-multilabel-task-fridge-items.yml

$schema: https://azuremlsdk2.blob.core.windows.net/preview/0.0.1/autoMLJob.schema.json
type: automl

experiment_name: dpv2-cli-automl-image-classification-multilabel-experiment
description: A multi-label Image classification job using fridge items dataset

compute: azureml:gpu-cluster

task: image_classification_multilabel
log_verbosity: debug
primary_metric: iou

target_column_name: label
training_data:
  path: data/training-mltable-folder
  type: mltable
validation_data:
  path: data/validation-mltable-folder
  type: mltable

limits:
  timeout_minutes: 60

image_model:
  early_stopping: True
  evaluation_frequency: 1

sweep:
  limits:
    max_trials: 10
    max_concurrent_trials: 2
  sampling_algorithm: random
  early_termination:
    type: bandit
    evaluation_interval: 2
    slack_factor: 0.2
    delay_evaluation: 6

search_space:
  - model_name: "vitb16r224"
    learning_rate: "uniform(0.005, 0.05)"
    number_of_epochs: "choice(15, 30)"
    grad_accumulation_step: "choice(1, 2)"
  - model_name: "seresnext"
    learning_rate: "uniform(0.005, 0.05)"
    validation_resize_size: "choice(288, 320, 352)"
    validation_crop_size: "choice(224, 256)"
    training_crop_size: "choice(224, 256)"