#!/bin/bash
#SBATCH --job-name=train_camera_obj
#SBATCH --partition=def
#SBATCH --account=prj-02-phai-lab
#SBATCH --nodes=1
#SBATCH --gres=gpu:4
#SBATCH --cpus-per-task=8
#SBATCH --mem=400G
#SBATCH --time=48:00:00
#SBATCH --output=/scratch/project/prj-02-phai-lab/lulin/jichen/outputs/train_camera_obj/slurm-%j.out
export WANDB_MODE=offline TOKENIZERS_PARALLELISM=false HF_HOME=/scratch/project/prj-02-phai-lab/lulin/jichen/.hf_home DS_ACCELERATOR=cuda
cd /scratch/project/prj-02-phai-lab/lulin/jichen/egohoi_extension
echo "node=$(hostname) gpus=4"; nvidia-smi -L 2>/dev/null | head -4
/scratch/project/prj-02-phai-lab/lulin/jichen/egohoi_extension/.venv/bin/python scripts/train.py --set camera_encoder.enable=true --set hand_encoder.enable=false --set obj_encoder.enable=true --set data.root=/scratch/project/prj-02-phai-lab/lulin/jichen/hot3d_data/outputs/hot3d_preprocessed --set data.val_root=/scratch/project/prj-02-phai-lab/lulin/jichen/hot3d_data/outputs/hot3d_preprocessed --set data.caption_annotations_path=/scratch/project/prj-02-phai-lab/lulin/jichen/Jichen/hot3d_qwen_captions.json --set data.num_workers=2 --set logging.tensorboard.enabled=true --set logging.tensorboard.log_dir=/scratch/project/prj-02-phai-lab/lulin/jichen/outputs/train_camera_obj/tensorboard --set logging.wandb.name=train_camera_obj --set output_dir=/scratch/project/prj-02-phai-lab/lulin/jichen/outputs/train_camera_obj --requested-gpus 4