plain_job_small_script.sh

#!/bin/bash
#SBATCH -N 1
#SBATCH -n 2
#SBATCH --gres=gpu:1
#SBATCH --mem=16G
#SBATCH -J no_colmap
#SBATCH --output=./slurm_outputs/no_colmap_depth_small%j.out



#2. Load Cuda12.1
module load cuda12.1/toolkit

#3. Load Cudnn 
# module load cudnn8.5-cuda11.7/8.5.0.96 

conda activate nerf
python3 train.py --config configs/gnt_plain_small.txt --train_scenes chair_small_dataset --eval_scenes chair_small_dataset