Add Qwen tail and chunk ablations
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user