Spaces:
Sleeping
Sleeping
File size: 543 Bytes
3e936b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | #!/bin/bash
set -e
GPU="${GPU:-0}"
SEED="${SEED:-0}"
LOGDIR="output/logs/long_1h"
mkdir -p "${LOGDIR}"
LOGFILE="${LOGDIR}/1h_gpu${GPU}_seed${SEED}.log"
OUT="output/long_1h/seed${SEED}"
echo "[$(date)] 1h streaming inference: gpu=${GPU} seed=${SEED} -> ${OUT}, log ${LOGFILE}"
CUDA_VISIBLE_DEVICES=${GPU} python inference/stream_long/inference_long_stream.py \
--config_path inference/stream_long/configs/cfg_1h.yaml \
--output_folder "${OUT}" \
--seed ${SEED} 2>&1 | tee "${LOGFILE}"
echo "[$(date)] 1h streaming inference complete"
|