name: ium_s434766-mlflow

docker_env:
  image: owczarczykp/ium_s434766:latest

entry_points:
  main:
    parameters:
      num_epochs: {type: int, default: 10}
      batch_size: {type: int, default: 16}
    command: "python3 lab8-mlflow.py -r {batch_size} {num_epochs}"