Xseg-Baseline / merge_pred.sh
YuqianFu's picture
Upload folder using huggingface_hub
944cdc2 verified
#!/bin/bash
#SBATCH --job-name=merge
#SBATCH --nodes=1 # Request 1 node
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=16 # Number of CPU cores (threads) per task
#SBATCH --mem-per-cpu=4G # Memory limit per CPU core (there is no --mem-per-task)
#SBATCH --time=48:00:00 # Job timeout
#SBATCH --output=merge_new4.log # Redirect stdout to a log file
#SBATCH --gpus-per-node=l4-24g:1
cd /home/yuqian_fu/Projects/ego-exo4d-relation/correspondence/XMem/scripts
srun --nodes "$SLURM_NNODES" --ntasks-per-node 1 -- \
mkenv -f /home/yuqian_fu/environment_xmem.yml -- \
sh -c "
python merge_pred_new.py --pred /data/work-gcp-europe-west4-a/yuqian_fu/Ego/inference_xmem_seg_ego_last_new/coco --input /data/work-gcp-europe-west4-a/yuqian_fu/Ego/data_segswap --gt /data/work-gcp-europe-west4-a/yuqian_fu/Ego/eval_gt_json/gt_mergeanno.json
"