name: 434788-mlflow
docker_env:
image: snowycocoon/ium_434788:4
entry_points:
main:
parameters:
batch_size: {type: int, default: 16}
epochs: {type: int, default: 15}
command: "python3 Zadanie_08_and_09_MLflow.py {batch_size} {epochs}"
