2022-05-15 12:32:27 +02:00
|
|
|
name: s478839
|
|
|
|
|
|
|
|
|
|
docker_env:
|
2022-05-15 13:11:42 +02:00
|
|
|
image: sebastianwalesa/ium:1.0
|
2022-05-15 12:32:27 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: float, default: 1000}
|
|
|
|
|
command: "python ml_pytorch_mlflow.py {epochs}"
|
