run.sh

#!/bin/bash
#SBATCH -p rivulet,brook
#SBATCH --gres=gpu:1
#SBATCH --mem=120G
#SBATCH --time=0-72:00:00
#SBATCH -o run_200.out
#SBATCH -e run_200.err

source /opt/csg/anaconda/etc/profile.d/conda.sh
conda activate /csghome/sm330/emb_env

export WANDB_API_KEY=$(cat /csghome/sm330/.netrc | grep password | awk '{print $2}')
export PYTHONPATH="/csghome/sm330/embed2heights-challenge-submission:$PYTHONPATH"

srun -u env PYTHONNOUSERSITE=1 \
    python src/train.py \
    --data-root data/npy \
    --decoder-type split \
    --batch-size 16 \
    --epochs 200 \
    --lr 1e-3 \
    --val-split 0.2 \
    --num-workers 32 \
    --oversample-buildings \
    --oversample-factor 5.0 \
    --experiment-name run_200 \
    --devices 1