---
common_options: &common_options
  data:
    throughput:
      reduction_type: "final"
      regexp: "throughput: *(.*?) samples\/sec"
    loss:
      reduction_type: "final"
      regexp: 'loss: *(\d*\.\d*)'
    compile_time:
      regexp: "Compiled model in *(.*?) secs"
  output:
    - [samples/sec, 'throughput']
    - [loss, 'loss']
    - [sec, 'compile_time']

config_options: &config_options
  requirements_path: requirements.txt

pytorch_minidalle_train_real_pod4:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod16:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD16
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod64:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD64
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod64_conv:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD64
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --wandb
      --wandb-project-name torch-miniDALL-E
      --wandb-run-name pytorch_minidalle_train_real_pod64_conv
      --checkpoint-output-dir output/ckpt
