ium_444354/MLproject
Adrian Charkiewicz 2fa5f88c56
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
myruns
2022-05-15 22:03:07 +02:00

13 lines
263 B
Plaintext

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