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}"