run_experiment.sh

#!/bin/bash

source ~/.bashrc
conda activate GraphMaker
export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:$LD_LIBRARY_PATH # Often needed to find libcusparse.so

DATA_DIR="$1"
N_SAMPLES="$2"
MODEL_NAME="$3"
if [[ -z "$DATA_DIR" || -z "$N_SAMPLES" || -z "$MODEL_NAME" ]]; then
  echo "Usage: sbatch run_experiment.sh <DATA_DIR> <N_SAMPLES> <MODEL_NAME>"
  exit 1
fi


SAMPLED_DIR="${DATA_DIR}/${MODEL_NAME}/sampled"
SYNTH_DIR="${DATA_DIR}/${MODEL_NAME}/synth"
MODEL_PATH="../botrgcn_cpts/${MODEL_NAME}"

mkdir -p "$SAMPLED_DIR" "$SYNTH_DIR"

CONSTANTS_FILE="./src/constants.py"
if [[ -f "$CONSTANTS_FILE" ]]; then
  WANDB_LINE="$(head -n 1 "$CONSTANTS_FILE")"
  if [[ "$WANDB_LINE" != WANDB_* ]]; then
    echo "[WARN] First line of constants.py doesn't look like WANDB_KEY; preserving it anyway."
  fi
else
  WANDB_LINE='WANDB_KEY = ""'
fi

esc() { printf '%s' "$1" | sed 's/\\/\\\\/g; s/"/\\"/g'; }

cat > "$CONSTANTS_FILE" <<EOF
$WANDB_LINE
DATA_DIR = "$(esc "$DATA_DIR")"
N_SAMPLES = $N_SAMPLES
MODEL_NAME = "$(esc "$MODEL_NAME")"
SAMPLED_DIR = "$(esc "$SAMPLED_DIR")"
SYNTH_DIR = "$(esc "$SYNTH_DIR")"
MODEL_PATH = "$(esc "$MODEL_PATH")"
EOF

cd src
python main.py