2009YU's picture
Release TMCRA Agent Memory v2026.07.18
490f3fe verified
Raw
History Blame Contribute Delete
761 Bytes
# reproduce_s500.sh changes into benchmarks/longmemeval before loading this
# file, so $PWD resolves to that directory.
DATA="$PWD/data/longmemeval_s_cleaned.json"
QIDS="$PWD/data/longmemeval_s500.qids.txt"
RUN="$PWD/runs/longmemeval_s500"
# TMCRA scorer weights are tracked by Git LFS in the repository-level model
# package. BGE snapshots are downloaded locally under this benchmark directory.
MODELS="$PWD/../../models/tmcra_v4_longmemeval_s500_20260715"
BGE_MODELS="$PWD/models/BAAI"
# Tune concurrency to provider rate limits and local memory. Device values are
# passed directly to PyTorch and therefore require a matching local runtime.
WRITER_CONCURRENCY=20
SLOW_CONCURRENCY=20
RETRIEVAL_DEVICE=cuda
GRAPH_DEVICE=cuda
ANSWER_WORKERS=12
JUDGE_WORKERS=8