Add final-layer mixture readout

run_extract_easycom_direct.sh

@@ -16,6 +16,7 @@ LOG_ROOT="${LOG_ROOT:-${PROJECT_ROOT}/logs/easycom_direct}"
 MEMORY_SECONDS="${MEMORY_SECONDS:-20}"
 NATIVE_BLOCK_SECONDS="${NATIVE_BLOCK_SECONDS:-2}"
 VIDEO_FPS="${VIDEO_FPS:-2}"
+READOUT_LAST_N_LAYERS="${READOUT_LAST_N_LAYERS:-1}"
 MODE="${MODE:-open-stream-replay}"
 DTYPE="${DTYPE:-bfloat16}"
 DECODER="${DECODER:-decord}"
@@ -72,6 +73,7 @@ run_one() {
     echo "target=${target}"
     echo "output=${output}"
     echo "memory_seconds=${MEMORY_SECONDS} video_fps=${VIDEO_FPS} mode=${MODE}"
+    echo "readout_last_n_layers=${READOUT_LAST_N_LAYERS}"
     echo "decoder=${DECODER} decoder_gpu_id=${DECODER_GPU_ID} frame_cache_size=${FRAME_CACHE_SIZE}"
 
     if ! CUDA_VISIBLE_DEVICES="${gpu}" PYTHONUNBUFFERED=1 \
@@ -84,6 +86,7 @@ run_one() {
         --memory-seconds "${MEMORY_SECONDS}" \
         --native-block-seconds "${NATIVE_BLOCK_SECONDS}" \
         --video-fps "${VIDEO_FPS}" \
+        --readout-last-n-layers "${READOUT_LAST_N_LAYERS}" \
         --frame-cache-size "${FRAME_CACHE_SIZE}" \
         --decoder "${DECODER}" \
         --decoder-gpu-id "${DECODER_GPU_ID}" \