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_t25_2_2025-08-28_01-29-51"
  #   wandb_run_name: "crow_t25_2_auk_dpo_r1"
  
  # - name: "round_1"
  #   script: "run_ipo_crow_bluejay_r1.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t25_2_2025-09-08_23-49-03"
  #   sft_loss_scale: 0.0
  #   wandb_run_name: "crow_t25_2_bjay_sft01_v42_r1"
  
  # - name: "round_2"
  #   script: "run_ipo_crow_bluejay_r2.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t25_2_bjay_sft01_v42_r1"
  #   sft_loss_scale: 0.0
  #   wandb_run_name: "crow_t25_2_bjay_sft01_v43_r2"
  
  # - name: "round_3"
  #   script: "run_ipo_crow_bluejay_r3.sh"
  #   nodes: 8
  #   model_cache_loss_name: "crow_t25_2_bjay_sft01_v43_r2"
  #   sft_loss_scale: 0.0
  #   wandb_run_name: "crow_t25_2_bjay_sft01_v44_r3"
  
  # - name: "round_4"
  #   script: "run_ipo_crow_bluejay_r4.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_r9"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v45_r4_bt8"
    
  # - name: "round_5"
  #   script: "run_ipo_crow_bluejay_r5.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_v45_r4_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v46_r5_bt8"
  
  # - name: "round_6"
  #   script: "run_ipo_crow_bluejay_r6.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v46_r5_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v47_r6_bt8"
  
  # - name: "round_7"
  #   script: "run_ipo_crow_bluejay_r7.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v47_r6_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v48_r7_bt8"
  
  # - name: "round_8"
  #   script: "run_ipo_crow_bluejay_r8.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v48_r7_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v59_r8_bt8"

  # - name: "round_9"
  #   script: "run_ipo_crow_bluejay_r9.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_v59_r8_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v60_r9_bt8"
    
  # - name: "round_10"
  #   script: "run_ipo_crow_bluejay_r10.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v60_r9_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v61_r10_bt8"
    
  # - name: "round_11"
  #   script: "run_ipo_crow_bluejay_r11.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v63_r10_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v63_r11_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_t25_2_bjay_dpo_v60_r11_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v45_r12_bt8"
  
  - name: "round_13"
    script: "run_ipo_crow_bluejay_r13.sh"  
    nodes: 16
    model_cache_loss_name: "crow_t25_2_bjay_dpo_v45_r12_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v46_r13_bt8"

  - name: "round_14"
    script: "run_ipo_crow_bluejay_r14.sh"
    nodes: 16
    model_cache_loss_name: "crow_t25_2_bjay_dpo_v46_r13_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v47_r14_bt8"

  - name: "round_15"
    script: "run_ipo_crow_bluejay_r15.sh"
    nodes: 16
    model_cache_loss_name: "crow_t25_2_bjay_dpo_v47_r14_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v48_r15_bt8"

  - name: "round_16"
    script: "run_ipo_crow_bluejay_r16.sh"
    nodes: 16
    model_cache_loss_name: "crow_t25_2_bjay_dpo_v48_r15_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v59_r16_bt8"

  - name: "round_17"
    script: "run_ipo_crow_bluejay_r17.sh"
    nodes: 16
    model_cache_loss_name: "crow_t25_2_bjay_dpo_v59_r16_bt8"
    wandb_run_name: "crow_t25_2_bjay_dpo_v60_r17_bt8"

  # - name: "round_18"
  #   script: "run_ipo_crow_bluejay_r18.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v60_r17_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v61_r18_bt8"

  # - name: "round_19"
  #   script: "run_ipo_crow_bluejay_r19.sh"
  #   nodes: 16
  #   model_cache_loss_name: "crow_t25_2_bjay_dpo_v61_r18_bt8"
  #   wandb_run_name: "crow_t25_2_bjay_dpo_v63_r19_bt8"