Blame view

scripts/evaluation/resume_coarse_fusion_tuning_knn_tail.sh 1.55 KB
935f6e1b   tangwang   coarse_rank 搜参结果 ...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
  #!/bin/bash
  
  set -euo pipefail
  
  if [ "$#" -lt 1 ]; then
    echo "usage: $0 <run_dir_or_name>" >&2
    exit 1
  fi
  
  cd "$(dirname "$0")/../.."
  source ./activate.sh
  
  TARGET="$1"
  
  if [ -d "${TARGET}" ]; then
    RUN_DIR="${TARGET}"
    RUN_NAME="$(basename "${RUN_DIR}")"
  else
    RUN_NAME="${TARGET}"
    RUN_DIR="artifacts/search_evaluation/tuning_runs/${RUN_NAME}"
  fi
  
  if [ ! -d "${RUN_DIR}" ]; then
    echo "run dir not found: ${RUN_DIR}" >&2
    exit 1
  fi
  
  DATASET_ID="${REPO_EVAL_DATASET_ID:-clothing_top771}"
  MAX_EVALS="${MAX_EVALS:-20}"
  BATCH_SIZE="${BATCH_SIZE:-2}"
  CANDIDATE_POOL_SIZE="${CANDIDATE_POOL_SIZE:-96}"
  RANDOM_SEED="${RANDOM_SEED:-20260424}"
  BATCH_EVAL_TIMEOUT_SEC="${BATCH_EVAL_TIMEOUT_SEC:-0}"
  
  LAUNCH_DIR="artifacts/search_evaluation/tuning_launches"
  mkdir -p "${LAUNCH_DIR}"
  LOG_PATH="${LAUNCH_DIR}/${RUN_NAME}.daemon.log"
  PID_PATH="${LAUNCH_DIR}/${RUN_NAME}.daemon.pid"
  CMD_PATH="${LAUNCH_DIR}/${RUN_NAME}.daemon.cmd"
  
  CMD=(
    bash
    scripts/evaluation/run_coarse_fusion_tuning_resilient.sh
    "${RUN_NAME}"
    "${DATASET_ID}"
    "${MAX_EVALS}"
    "${BATCH_SIZE}"
    "${CANDIDATE_POOL_SIZE}"
    "${RANDOM_SEED}"
    "${RUN_DIR}/search_space.yaml"
    ""
    "${RUN_DIR}"
  )
  
  export BATCH_EVAL_TIMEOUT_SEC
  
  printf '%q ' "${CMD[@]}" > "${CMD_PATH}"
  printf '\n' >> "${CMD_PATH}"
  
  setsid "${CMD[@]}" > "${LOG_PATH}" 2>&1 < /dev/null &
  PID=$!
  echo "${PID}" > "${PID_PATH}"
  
  echo "run_name=${RUN_NAME}"
  echo "pid=${PID}"
  echo "log=${LOG_PATH}"
  echo "pid_file=${PID_PATH}"
  echo "cmd_file=${CMD_PATH}"
  echo "run_dir=${RUN_DIR}"
  echo
  echo "tail -f ${LOG_PATH}"
  echo "cat ${RUN_DIR}/leaderboard.csv"