---
common_options: &common_options
  data:
    throughput:
      skip: 1
      regexp: 'num_frame/sec *(.*?) '
  output:
    - [frames/sec, 'throughput']

config_options: &config_options
  requirements_path: requirements.txt

pytorch_fastpitch_train_gen_pod16:
  <<: [*common_options, *config_options]
  description: PyTorch FastPitch throughput benchmark using generated data.
  cmd: >-
    python train.py
      --checkpoint-dir ./
      --epochs 1
      --learning-rate 0.1
      --batch-size 2
      --dur-predictor-loss-scale 0.1
      --pitch-predictor-loss-scale 0.1
      --gradient-accumulation 16
      --replication-factor 8
      --num-dataloader-workers 64
      --generated-data
