chain_name: "Aura G chain"
description: "Aura G chain, 9 steps."

# Global settings
settings:
  base_dir: "/home/tony/Work/tony/slurm/bluejay"
  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_bluejay_r1.sh"
  #   nodes: 4
  #   model_cache_loss_name: "6b_crow_25_0"
  #   wandb_run_name: "dpo_6b_base0828_crow_25_r1"

  # - name: "round_2"
  #   script: "run_ipo_bluejay_r2_task.sh"
  #   nodes: 8
  #   model_cache_loss_name: "6b_crow_25_1"
  #   wandb_run_name: "dpo_6b_base0828_crow_25_r2"
    
  - name: "round_3"
    script: "run_ipo_bluejay_r3_exp.sh"  
    nodes: 16
    model_cache_loss_name: "6b_crow_25_2"
    wandb_run_name: "dpo_6b_base0828_crow_25_r3"

  - name: "round_4"
    script: "run_ipo_bluejay_r4_mix.sh"
    nodes: 16
    model_cache_loss_name: "6b_crow_25_3"
    wandb_run_name: "dpo_6b_base0828_crow_25_r4"
    
  - name: "round_5"
    script: "run_ipo_bluejay_r5_mix_2.sh"  
    nodes: 10
    model_cache_loss_name: "6b_crow_25_4"
    wandb_run_name: "dpo_6b_base0828_crow_25_r5"

  - name: "round_6"
    script: "run_ipo_bluejay_r6_auk.sh"
    nodes: 16
    model_cache_loss_name: "6b_crow_25_5"
    wandb_run_name: "dpo_6b_base0828_crow_25_r6"
    
  - name: "round_7"
    script: "run_ipo_bluejay_r7_auk_2.sh"  
    nodes: 16
    model_cache_loss_name: "6b_crow_25_6"
    wandb_run_name: "dpo_6b_base0828_crow_25_r7"
    
  - name: "round_8"
    script: "run_ipo_bluejay_r8_auk_3.sh"  
    nodes: 16
    model_cache_loss_name: "6b_crow_25_7"
    wandb_run_name: "dpo_6b_base0828_crow_25_r8"
    
  - name: "round_9"
    script: "run_ipo_bluejay_r9_auk_4.sh"  
    nodes: 16
    model_cache_loss_name: "6b_crow_25_8"
    wandb_run_name: "dpo_6b_base0828_crow_25_r9"
    
  # - name: "round_10"
  #   script: "run_ipo_bluejay_r10_bluejay_1.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_20_r13_v25"
  #   wandb_run_name: "dpo_6b_base0828_bjay_20_r14_v24"
    
  # - name: "round_11"
  #   script: "run_ipo_bluejay_r11_bluejay_2.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_20_r14_v24"
  #   wandb_run_name: "dpo_6b_base0828_bjay_20_r15_v25"

  # - name: "round_12"
  #   script: "run_ipo_bluejay_r12_bluejay_3.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_27_d43_6"
  #   wandb_run_name: "dpo_6b_base0828_bjay_27_d43_r7_v26"
    
  # - name: "round_13"
  #   script: "run_ipo_bluejay_r13_bluejay_4.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_27_d43_7"
  #   wandb_run_name: "dpo_6b_base0828_bjay_27_d43_r8_v27"