name: mnist-autolog-example

python_env: python_env.yaml

entry_points:
  main:
    parameters:
      max_epochs: {type: int, default: 5}
      devices : {type: str, default: "auto"}
      strategy: {type str, default: "auto"}
      accelerator: {type str, default: "auto"}
      batch_size: {type: int, default: 64}
      num_workers: {type: int, default: 3}
      learning_rate: {type: float, default: 0.001}

    command: |
          python mnist_autolog_example.py \
            --trainer.max_epochs={max_epochs} \
            --trainer.devices={devices} \
            --trainer.strategy={strategy} \
            --trainer.accelerator={accelerator} \
            --data.batch_size={batch_size} \
            --data.num_workers={num_workers} \
            --model.learning_rate={learning_rate}
