diffusion_baseline / scripts /smoke_all.sh
jyliuAI's picture
Add files using upload-large-folder tool
840c3a4 verified
Raw History Blame Contribute Delete
1.19 kB
#!/usr/bin/env bash
# Server-side one-sample smoke pass. Its sample_count=1 identity cannot satisfy
# the production 1,024-sample aggregation contract.
set -uo pipefail
SCRIPT_DIR="$(CDPATH= cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd -P)"
DLB_ROOT="${DLB_ROOT:-$(CDPATH= cd -- "$SCRIPT_DIR/.." && pwd -P)}"
PYTHON_BIN="${DLB_PYTHON:-python}"
MATRIX="${DLB_ROOT}/results/matrix/smoke.tsv"
if [[ "${1:-}" == "--help" ]]; then
printf 'Usage: %s\n' "$0"
exit 0
fi
mkdir -p "$DLB_ROOT/results/matrix" "$DLB_ROOT/results/logs"
PYTHONPATH="$DLB_ROOT/src${PYTHONPATH:+:$PYTHONPATH}" "$PYTHON_BIN" -m dlb.matrix \
--root "$DLB_ROOT" --output "$MATRIX" --sample-count 1 --seed 42 || exit $?
status=0
while IFS=$'\t' read -r task_id category model dataset steps sample_count seed environment adapter source provenance sample_dir metrics_path timing_path; do
[[ "$task_id" == "task_id" || -z "$task_id" ]] && continue
echo "SMOKE $task_id"
"$SCRIPT_DIR/run_one.sh" --model "$model" --dataset "$dataset" --steps "$steps" \
--num-samples 1 --seed "$seed" --results-root "$DLB_ROOT/results/smoke"
if (($? != 0)); then status=1; fi
done < <(tail -n +3 "$MATRIX")
exit "$status"