run_nvdiffrecmc_slurm.sh

#!/bin/bash

source ~/.bashrc
actenv py311

DATA_ROOT=/svl/data/Aria_DTC/ORB_eval_format_spherified
CODE_ROOT=/svl/u/yzzhang/projects/aria/imageint
THIRD_PARTY_CODE_ROOT=${CODE_ROOT}/imageint/third_party/nvdiffrecmc

SCENE=scene000_000_Birdhouse
EXP_ID=1110

COMMAND="python train.py \
--config ${CODE_ROOT}/configs/nvdiffrecmc.json \
-rm ${DATA_ROOT}/$SCENE/final_output/blender_format_HDR \
-o $SCENE/${EXP_ID}"

python -m tu.sbatch.sbatch_sweep --time 72:00:00 \
--proj_dir ${THIRD_PARTY_CODE_ROOT} --conda_env dmodel2 \
--console_output_dir /svl/u/yzzhang/projects/aria/sbatch_console_outputs/ \
--job nvdiffrecmc_${SCENE}_${EXP_ID} --command "$COMMAND" \
--partition viscam --cpus_per_task 8 --mem 20G \
exclude=viscam1,viscam2,viscam7,svl[1-6]