gpu/single-gpu/home_benjamin_hdbasin/code/scripts/run_medium.sh

#!/usr/bin/env bash
set -uo pipefail

OUT_ROOT="${1:-runs/isobasinflow_medium_$(date -u +%Y%m%d_%H%M%S)}"
export SMOKE_SEEDS="${MEDIUM_SYNTH_SEEDS:-5}"
export SMOKE_BUDGET="${MEDIUM_SYNTH_BUDGET:-32}"
export MARKET_SMOKE_SEEDS="${MEDIUM_MARKET_SEEDS:-3}"
export MARKET_SMOKE_SEED_OFFSET="${MEDIUM_MARKET_SEED_OFFSET:-8200}"
export MARKET_SMOKE_TEXT_DATASETS="${MEDIUM_TEXT_DATASETS:-sst2,ag_news,tweet_sentiment,emotion}"
export MARKET_SMOKE_IMAGE_DATASETS="${MEDIUM_IMAGE_DATASETS:-cifar10,fashion_mnist}"
export MARKET_SMOKE_TEXT_BUDGET="${MEDIUM_TEXT_BUDGET:-6}"
export MARKET_SMOKE_IMAGE_BUDGET="${MEDIUM_IMAGE_BUDGET:-6}"
export MARKET_SMOKE_TEXT_TRAIN_SIZE="${MEDIUM_TEXT_TRAIN_SIZE:-128}"
export MARKET_SMOKE_TEXT_VAL_SIZE="${MEDIUM_TEXT_VAL_SIZE:-128}"
export MARKET_SMOKE_IMAGE_TRAIN_SIZE="${MEDIUM_IMAGE_TRAIN_SIZE:-512}"
export MARKET_SMOKE_IMAGE_VAL_SIZE="${MEDIUM_IMAGE_VAL_SIZE:-256}"
export MARKET_SMOKE_IMAGE_EPOCHS="${MEDIUM_IMAGE_EPOCHS:-1}"

RESET_RUN_LOG="${RESET_RUN_LOG:-1}" bash scripts/run_smoke.sh "${OUT_ROOT}"