cli-automl-image-classification-multiclass-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-experiment
description: A multi-class Image classification job using fridge items dataset

compute: azureml:gpu-cluster

task: image_classification
log_verbosity: debug
primary_metric: accuracy

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: "choice('vitb16r224', 'vits16r224')"
    learning_rate: "uniform(0.001, 0.01)"
    number_of_epochs: "choice(15, 30)"
  - model_name: "choice('seresnext', 'resnet50')"
    learning_rate: "uniform(0.001, 0.01)"
    layers_to_freeze: "choice(0, 2)"