name: s464980

docker_env:
  image:  s464980-mlflow

entry_points:
  main:
    parameters:
      epochs: float
    command: "python train.py --epochs {epochs}"