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"