#!/bin/bash

python trainer.py \
    --dataset $DATASET \
    --run $RUN \
    --batch-size $BATCH_SIZE \
    --mode $MODE \
    --ood-test $OOD_TEST \
    --corrupt-test $CORRUPT_TEST \
    --gpu $GPU \
    --save-best-val $SAVE_BEST_VAL \
    --model $MODEL \
    --p $P \
    --filterwise-dropout $FILTERWISE_DROPOUT \
    --data-root $DATADIR
