ium_444354/MLproject

13 lines
262 B
Plaintext
Raw Normal View History

2022-05-15 19:20:14 +02:00
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"