chain_name: "Crow DPO Chain"
description: "A Crow DPO Chain"

# Global settings
settings:
  base_dir: "/home/tony/Work/tony/slurm/crow"
  log_dir: "/home/tony/slurm/logs"
  
  # SLURM defaults - no partition specified
  slurm_defaults:
    time: "48:00:00"
    ntasks_per_node: 8
    cpus_per_task: 4
    gpus_per_node: 8

# Job definitions
jobs:
  # - name: "round_1"
  #   script: "run_ipo_crow_auk_dpo_r1.sh"
  #   nodes: 4
  #   model_cache_loss_name: "6b_crow_t14_2025-08-28_01-29-51"
  #   wandb_run_name: "crow_t14_auk_dpo_r1"
  
  # - name: "round_1"
  #   script: "run_ipo_crow_bluejay_r1.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t14_2025-08-28_01-30-09"
  #   sft_loss_scale: 0.01
  #   wandb_run_name: "crow_t14_bjay_sft01_v42_r1"
  
  # - name: "round_2"
  #   script: "run_ipo_crow_bluejay_r2.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t14_bjay_sft01_v42_r1"
  #   sft_loss_scale: 0.01
  #   wandb_run_name: "crow_t14_bjay_sft01_v43_r2"
  
  # - name: "round_3"
  #   script: "run_ipo_crow_bluejay_r3.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t14_bjay_sft01_v43_r2"
  #   sft_loss_scale: 0.01
  #   wandb_run_name: "crow_t14_bjay_sft01_v44_r3"
  
  # - name: "round_4"
  #   script: "run_ipo_crow_bluejay_r4.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_sft01_v44_r3"
  #   wandb_run_name: "crow_t14_bjay_dpo_v45_r4_bt8"
    
  # - name: "round_5"
  #   script: "run_ipo_crow_bluejay_r5.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_v45_r4_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v46_r5_bt8"
  
  # - name: "round_6"
  #   script: "run_ipo_crow_bluejay_r6.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v46_r5_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v47_r6_bt8"
  
  # - name: "round_7"
  #   script: "run_ipo_crow_bluejay_r7.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v47_r6_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v48_r7_bt8"
  
  # - name: "round_8"
  #   script: "run_ipo_crow_bluejay_r8.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v48_r7_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v59_r8_bt8"

  # - name: "round_9"
  #   script: "run_ipo_crow_bluejay_r9.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_v59_r8_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v60_r9_bt8"
    
  # - name: "round_16"
  #   script: "run_ipo_crow_bluejay_r10.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v60_r15_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v61_r16_bt8"
    
  # - name: "round_17"
  #   script: "run_ipo_crow_bluejay_r11.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v62_r16_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v62_r17_bt8"
  
  # # this is the 2nd round, we increased beta to 40
  # - name: "round_12"
  #   script: "run_ipo_crow_bluejay_r12.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v62_r11_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v45_r12_bt8"
  
  # - name: "round_13"
  #   script: "run_ipo_crow_bluejay_r13.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v45_r12_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v46_r13_bt8"

  # - name: "round_14"
  #   script: "run_ipo_crow_bluejay_r14.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v46_r13_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v47_r14_bt8"

  # - name: "round_15"
  #   script: "run_ipo_crow_bluejay_r15.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v47_r14_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v48_r15_bt8"

  # - name: "round_16"
  #   script: "run_ipo_crow_bluejay_r16.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v48_r15_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v59_r16_bt8"

  # - name: "round_17"
  #   script: "run_ipo_crow_bluejay_r17.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t14_bjay_dpo_v59_r16_bt8"
  #   wandb_run_name: "crow_t14_bjay_dpo_v60_r17_bt8"

  - name: "round_18"
    script: "run_ipo_crow_bluejay_r18.sh"
    nodes: 16
    model_cache_loss_name: "crow_t14_bjay_dpo_v62_r17_bt8"
    wandb_run_name: "crow_t14_bjay_dpo_v61_r18_bt8"

  - name: "round_19"
    script: "run_ipo_crow_bluejay_r19.sh"
    nodes: 16
    model_cache_loss_name: "crow_t14_bjay_dpo_v61_r18_bt8"
    wandb_run_name: "crow_t14_bjay_dpo_v62_r19_bt8"
