inference/run_interactive.sh

#!/bin/bash
set -e

GPU="${GPU:-0}"
SEED="${SEED:-1}"
CONFIG="${CONFIG:-configs/echo_infinity-long_interactive.yaml}"
OUT="${OUT:-output/echo_infinity_long_interactive/seed${SEED}}"
LOGDIR="output/logs/interactive"
mkdir -p "${LOGDIR}"
LOGFILE="${LOGDIR}/interactive_gpu${GPU}_seed${SEED}.log"

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

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

echo "[$(date)] interactive inference complete"