#!/bin/bash

python trainer.py \
    --dataset $DATASET \
    --run $RUN \
    --batch-size $BATCH_SIZE \
    --n-way $WAY \
    --k-shot $SHOT \
    --mode $MODE \
    --ood-test $OOD_TEST \
    --corrupt-test $CORRUPT_TEST \
    --gpu $GPU \
    --train-query-shots $TRAIN_QUERY_SHOTS \
    --val-query-shots $VAL_QUERY_SHOTS \
    --save-best-val $SAVE_BEST_VAL \
    --data-root $DATADIR
