name: GPU Training Example

python_env: python_env.yaml

entry_points:
  main:
    command: "python train.py"
