Add Qwen tail and chunk ablations

This commit is contained in:
Xin Wang
2026-09-17 02:32:08 +08:00
parent a183fe5944
commit a9f748d6a6
7 changed files with 1219 additions and 32 deletions
+9 -5
View File
@@ -16,16 +16,20 @@ LOG_ROOT="${LOG_ROOT:-${PROJECT_ROOT}/logs/easycom_direct_smoke}"
SESSION_ID="${SESSION_ID:-1}"
GPU="${GPU:-0}"
MODE="${MODE:-open-stream-replay}"
# Short smoke: END=4. Full session smoke: END="" (empty).
END="${END:-4}"
MEMORY_SECONDS="${MEMORY_SECONDS:-20}"
# The historical short smoke used persistent KV (0). Override this to 8 for
# the matched bounded-memory chunk ablation.
SHORT_MEMORY_SECONDS="${SHORT_MEMORY_SECONDS:-0}"
VIDEO_FPS="${VIDEO_FPS:-2}"
session="Session_${SESSION_ID}"
session_dir="${VIDEO_ROOT}/${session}"
target="${TARGET_ROOT}/${session}.npy"
output="${FEATURE_ROOT}/${session}_gpu${GPU}.pt"
log="${LOG_ROOT}/${session}_gpu${GPU}.log"
output="${FEATURE_ROOT}/${session}_${MODE}_gpu${GPU}.pt"
log="${LOG_ROOT}/${session}_${MODE}_gpu${GPU}.log"
mkdir -p "${FEATURE_ROOT}" "${LOG_ROOT}"
cd "${PROJECT_ROOT}"
@@ -36,13 +40,13 @@ extra_args=()
if [[ -n "${END}" ]]; then
# Bounded smoke: do not pass --target (length would conflict with --end).
extra_args+=(--end "${END}")
MEMORY_SECONDS=0
MEMORY_SECONDS="${SHORT_MEMORY_SECONDS}"
else
[[ -s "${target}" ]] || { echo "Missing: ${target}" >&2; exit 1; }
extra_args+=(--target "${target}")
fi
echo "GPU=${GPU} session=${session} end=${END:-full} memory=${MEMORY_SECONDS}"
echo "GPU=${GPU} session=${session} mode=${MODE} end=${END:-full} memory=${MEMORY_SECONDS}"
echo "log -> ${log}"
{
@@ -52,7 +56,7 @@ echo "log -> ${log}"
--model-id "${MODEL}" \
--easycom-session-dir "${session_dir}" \
--use-audio-in-video \
--mode open-stream-replay \
--mode "${MODE}" \
--memory-seconds "${MEMORY_SECONDS}" \
--video-fps "${VIDEO_FPS}" \
--dtype bfloat16 \