ium_478839/MLproject

11 lines
196 B
Plaintext

name: s478839
docker_env:
image: sebastianwalesa/ium:1.0
entry_points:
main:
parameters:
epochs: {type: float, default: 1000}
command: "python ml_pytorch_mlflow.py {epochs}"