46 lines
1.4 KiB
Bash
46 lines
1.4 KiB
Bash
|
#!/bin/bash
|
||
|
|
# 登录 HF → 下载 ShieldGemma-2B → 运行评估，全程写入日志
|
||
|
|
set -e
|
||
|
|
|
||
|
|
PROJ=/root/siton-data-2849d4ce327c4ccfb233ce33868fe7fe/zsy/CompanionGuard-RL
|
||
|
|
MODEL_DIR=/root/siton-data-2849d4ce327c4ccfb233ce33868fe7fe/zsy/shieldgemma-2b
|
||
|
|
PY=/opt/conda/envs/dlapo-py310-cu128/bin/python
|
||
|
|
HF=/opt/conda/envs/dlapo-py310-cu128/bin/hf
|
||
|
|
LOG=$PROJ/experiments/run_shieldgemma2b_$(date +%Y%m%d_%H%M%S).log
|
||
|
|
|
||
|
|
mkdir -p $PROJ/experiments
|
||
|
|
|
||
|
|
# 从这里开始把所有输出重定向到日志文件
|
||
|
|
exec > "$LOG" 2>&1
|
||
|
|
|
||
|
|
echo "=== $(date) START ==="
|
||
|
|
echo "PROJ=$PROJ"
|
||
|
|
echo "MODEL_DIR=$MODEL_DIR"
|
||
|
|
|
||
|
|
# 代理（服务器无外网）
|
||
|
|
export http_proxy=http://127.0.0.1:7890
|
||
|
|
export https_proxy=http://127.0.0.1:7890
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "--- [1/3] HuggingFace Login ---"
|
||
|
|
$HF auth login --token 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
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "--- [2/3] Downloading google/shieldgemma-2b ---"
|
||
|
|
$HF download google/shieldgemma-2b \
|
||
|
|
--local-dir "$MODEL_DIR"
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "--- [3/3] Running evaluation on CompanionRisk-Bench test set ---"
|
||
|
|
cd "$PROJ"
|
||
|
|
export PYTHONPATH="$PROJ"
|
||
|
|
CUDA_VISIBLE_DEVICES=0 $PY scripts/eval_sota_baselines.py \
|
||
|
|
--model shieldgemma2b \
|
||
|
|
--model-path "$MODEL_DIR" \
|
||
|
|
--test-data data/processed/CompanionRisk-Bench/test.jsonl \
|
||
|
|
--output experiments/eval_sota_shieldgemma2b.json
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "=== $(date) DONE ==="
|
||
|
|
echo "Result: $PROJ/experiments/eval_sota_shieldgemma2b.json"
|