cd /scratch/phys/project/sin/hackathon
srun -p gpu-debug --gres=gpu:1 --time=00:30:00 --mem=16G --pty bash
module load mamba gcc cuda
conda activate /scratch/phys/project/sin/hackathon/.conda_envs/hack
python train.py --config configs/synth_3D.yaml --cuda_visible_device 0 --nproc_per_node 1