shinka-backup / scripts /dev /2_test_quick.sh
JustinTX's picture
Add files using upload-large-folder tool
3f6526a verified
#!/bin/bash
# Quick test: 3 generations with eval service and WandB
#
# Prerequisites: Eval service should be running
# bash scripts/dev/start_eval_server.sh
set -e
# ============================================================================
# Configuration
# ============================================================================
EXPERIMENT_NAME_PREFIX="debug"
NUM_GENERATIONS=10
MAX_PARALLEL_JOBS=2
META_INTERVAL=20
EXPERIMENT_NAME=${EXPERIMENT_NAME_PREFIX}_gen${NUM_GENERATIONS}
TASK="circle_packing"
LLM_MODELS="native-gemini-2.5-flash native-gemini-2.5-pro"
LLM_SELECTION="ucb1"
LLM_TEMPERATURES="0.5 0.7 1.0"
USE_EVAL_SERVICE="--use-eval-service"
EVAL_SERVICE_URL="http://localhost:8765"
EVAL_TRIGGER_MODE="periodic"
EVAL_TRIGGER_INTERVAL=20
USE_WANDB="--use-wandb"
WANDB_PROJECT="ev2"
WANDB_TAGS="${TASK} eval-service"
# ============================================================================
# Run Experiment
# ============================================================================
python scripts/dev/run_experiment.py \
--experiment-name "$EXPERIMENT_NAME" \
--num-generations "$NUM_GENERATIONS" \
--max-parallel-jobs "$MAX_PARALLEL_JOBS" \
--meta-interval "$META_INTERVAL" \
--task "$TASK" \
--llm-models $LLM_MODELS \
--llm-selection "$LLM_SELECTION" \
--llm-temperatures $LLM_TEMPERATURES \
$USE_EVAL_SERVICE \
--eval-service-url "$EVAL_SERVICE_URL" \
--eval-trigger-mode "$EVAL_TRIGGER_MODE" \
--eval-trigger-interval "$EVAL_TRIGGER_INTERVAL" \
$USE_WANDB \
--wandb-project "$WANDB_PROJECT" \
--wandb-tags $WANDB_TAGS \
--verbose