Skip to content

Commit 21a345e

Browse files
authored
Merge branch 'dev' into mac-win-adapt
2 parents e5067d1 + 6148813 commit 21a345e

File tree

2 files changed

+44
-1
lines changed

2 files changed

+44
-1
lines changed

.gitignore

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ evaluation/data/langmemeval
1010
evaluation/*tmp/
1111
evaluation/results
1212
evaluation/.env
13-
evaluation/scripts/*.sh
1413
evaluation/configs/*
1514
.env
1615

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
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!"

0 commit comments

Comments
 (0)