inference/run_inference.sh

#!/bin/bash
set -e

GPU="${GPU:-0}"
SEED="${SEED:-0}"
CONFIG="${CONFIG:-configs/echo_infinity-long_inference.yaml}"
OUT="${OUT:-output/echo_infinity_long_inference/seed${SEED}}"
LOGDIR="output/logs/inference"
mkdir -p "${LOGDIR}"
LOGFILE="${LOGDIR}/inference_gpu${GPU}_seed${SEED}.log"

echo "[$(date)] inference: gpu=${GPU} seed=${SEED} config=${CONFIG} -> ${OUT}"

CUDA_VISIBLE_DEVICES=${GPU} torchrun --nproc_per_node=1 --master_port=29500 \
  inference/inference.py \
    --config_path "${CONFIG}" \
    --output_folder "${OUT}" \
    --seed ${SEED} 2>&1 | tee "${LOGFILE}"

echo "[$(date)] inference complete"