name: tutorial
# conda_env: conda.yaml
docker_env:
image: ksero/ium:mlflow
entry_points:
main:
parameters:
epochs: {type: float, default: 1500}
command: "python ./pytorch/pytorch.py {epochs}"
test:
command: "python ./evaluation.py"