#!/bin/bash
#SBATCH --job-name=sam2
#SBATCH --nodes=1               # Request 1 node
#SBATCH --ntasks-per-node=1
#SBATCH --gpus-per-task=a100-40g:1
#SBATCH --cpus-per-gpu=8       # Number of CPU cores (threads) per task
#SBATCH --mem-per-gpu=40G        # Memory limit per CPU core (there is no --mem-per-task)
#SBATCH --time=24:00:00         # Job timeout
#SBATCH --output=sam2_result.log     # Redirect stdout to a log file
#SBATCH --nodelist=gcp-us-0


cd /home/yuqian_fu/Projects/sam2


srun --nodes "$SLURM_NNODES" --ntasks-per-node 1 -- \
mkenv -f sam2.yml -- \
sh -c "
python vos_inference.py --output_mask_dir /data/work-gcp-europe-west4-a/yuqian_fu/Ego/sam_result
"