name: s487187-training

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