#! /bin/bash
#SBATCH --nodes=1
#SBATCH -J mednist_train
#SBATCH -c 4
#SBATCH --gres=gpu:1
#SBATCH --time=2:00:00
#SBATCH -p small

set -v

# change this if run submitted from a different directory
export BUNDLE="$(pwd)/.."

# change this to load a checkpoint instead of started from scratch
CKPT=none

CONFIG="$BUNDLE/configs/train.yaml"

# change this to point to where MedNIST is located
DATASET="$(pwd)"

# it's useful to include the configuration in the log file
cat "$BUNDLE/configs/train.yaml"

python -m monai.bundle run training \
    --meta_file "$BUNDLE/configs/metadata.json" \
    --config_file "$CONFIG" \
    --logging_file "$BUNDLE/configs/logging.conf" \
    --bundle_root "$BUNDLE" \
    --dataset_dir "$DATASET"