File tree Expand file tree Collapse file tree 2 files changed +44
-1
lines changed Expand file tree Collapse file tree 2 files changed +44
-1
lines changed Original file line number Diff line number Diff line change @@ -10,7 +10,6 @@ evaluation/data/langmemeval
1010evaluation /* tmp /
1111evaluation /results
1212evaluation /.env
13- evaluation /scripts /* .sh
1413evaluation /configs /*
1514.env
1615
Original file line number Diff line number Diff line change 1+ #! /bin/bash
2+
3+ # Common parameters for all scripts
4+ LIB=" memos"
5+ VERSION=" 063001"
6+ WORKERS=10
7+ TOPK=20
8+
9+ echo " Running locomo_ingestion.py..."
10+ CUDA_VISIBLE_DEVICES=0 python scripts/locomo/locomo_ingestion.py --lib $LIB --version $VERSION --workers $WORKERS
11+ if [ $? -ne 0 ]; then
12+ echo " Error running locomo_ingestion.py"
13+ exit 1
14+ fi
15+
16+ echo " Running locomo_search.py..."
17+ CUDA_VISIBLE_DEVICES=0 python scripts/locomo/locomo_search.py --lib $LIB --version $VERSION --top_k $TOPK --workers $WORKERS
18+ if [ $? -ne 0 ]; then
19+ echo " Error running locomo_search.py"
20+ exit 1
21+ fi
22+
23+ echo " Running locomo_responses.py..."
24+ python scripts/locomo/locomo_responses.py --lib $LIB --version $VERSION
25+ if [ $? -ne 0 ]; then
26+ echo " Error running locomo_responses.py."
27+ exit 1
28+ fi
29+
30+ echo " Running locomo_eval.py..."
31+ python scripts/locomo/locomo_eval.py --lib $LIB --version $VERSION --workers $WORKERS --num_runs 3
32+ if [ $? -ne 0 ]; then
33+ echo " Error running locomo_eval.py"
34+ exit 1
35+ fi
36+
37+ echo " Running locomo_metric.py..."
38+ python scripts/locomo/locomo_metric.py --lib $LIB --version $VERSION
39+ if [ $? -ne 0 ]; then
40+ echo " Error running locomo_metric.py"
41+ exit 1
42+ fi
43+
44+ echo " All scripts completed successfully!"
You can’t perform that action at this time.
0 commit comments