2022-05-15 19:20:14 +02:00
|
|
|
name: tutorial
|
|
|
|
|
# conda_env: conda.yaml
|
|
|
|
|
|
|
|
|
|
docker_env:
|
2022-05-16 00:12:57 +02:00
|
|
|
image: ksero/ium:mlflow
|
2022-05-15 19:20:14 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: float, default: 1500}
|
|
|
|
|
command: "python ./pytorch/pytorch.py {epochs}"
|
|
|
|
|
test:
|
|
|
|
|
command: "python ./evaluation.py"
|
