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
|