File size: 713 Bytes
48ecd01
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/usr/bin/env bash
# Usage: bash scripts/run_eval.sh <checkpoint_dir>
# Example: bash scripts/run_eval.sh checkpoints/korean_1b_fp8_run1/checkpoint-0200000
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"

CHECKPOINT="${1:?Usage: bash scripts/run_eval.sh <checkpoint_dir>}"

echo "=== Perplexity Evaluation ==="
python "$PROJECT_DIR/eval/perplexity.py" \
  --checkpoint "$CHECKPOINT" \
  --data "$PROJECT_DIR/data/korean_val.bin" \
  --device cuda:0

echo ""
echo "=== Text Generation ==="
python "$PROJECT_DIR/eval/generate.py" \
  --checkpoint "$CHECKPOINT" \
  --prompt "์•ˆ๋…•ํ•˜์„ธ์š”, ์ €๋Š”" \
  --max_new_tokens 200 \
  --device cuda:0