#!/bin/bash -l
#SBATCH --job-name=om-cmp-opt
#SBATCH --account=AIRR-P51-DAWN-GPU
#SBATCH --partition=pvc9
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH --time=02:00:00
#SBATCH --output=Logs/compare_opt_%j.out
#SBATCH --error=Logs/compare_opt_%j.err

. /etc/profile.d/modules.sh
module purge
module load rhel9/default-dawn

source ~/miniconda3/etc/profile.d/conda.sh
conda activate ~/rds/rds-airr-p51-TWhPgQVLKbA/Env/pub_env/pytorch-xpu

export TMPDIR=/home/dn-zhen2/rds/rds-airr-p51-TWhPgQVLKbA/archive/tmp
mkdir -p "$TMPDIR"

cd /rds/project/rds-TWhPgQVLKbA/Code/OmniMorph

python -u tests/compare_3modes_speed.py --pipeline opt --device xpu --steps 10 --batchsize 1 --img-size 128