#!/bin/bash
#SBATCH --job-name="dpo"
#SBATCH --nodes=2
#SBATCH --ntasks-per-node=8
#SBATCH --cpus-per-task=4           # Number of cores per tasks
#SBATCH --gres=gpu:8                 # Number of gpus
#SBATCH --output=/app/suno/slurm/logs/run_%x_%j.txt   # Set this dir where you want slurm outs to go
#SBATCH --error=/app/suno/slurm/logs/run_%x_%j_err.txt    # Set this dir where you want slurm outs to go

# setup the environment using the script we created before

# srun -K1 run_auk_t1.sh  # 4 nodes
# srun -K1 run_auk_3b_og_r1.sh  # 2 nodes
srun -K1 run_auk_3b_og_r2.sh  # 2 nodes