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"