ium_444354/MLproject
Adrian Charkiewicz bdf10f8606 lab8
2022-05-15 19:20:14 +02:00

13 lines
262 B
Plaintext

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"