Download scripts/test/train_opd.sh from Jack04810/agentic-rl-main: direct link, hf CLI and curl.
- Browser
- Download file 854 Bytes
-
https://huggingface.co/Jack04810/agentic-rl-main/resolve/main/scripts/test/train_opd.sh
- Command line
-
hf download hf://Jack04810/agentic-rl-main/scripts/test/train_opd.sh
-
curl -L -o train_opd.sh https://huggingface.co/Jack04810/agentic-rl-main/resolve/main/scripts/test/train_opd.sh
854 Bytes
| # Fast baseline: cross-model OPD (7B teacher + 0.5B student) with DeepSpeed. | |
| # DyME-aligned routing (teacher-probe gated OPD, no embedded cold-start, no Visual Supervision). | |
| set -euo pipefail | |
| TEST_DIR="$(cd "$(dirname "$0")" && pwd)" | |
| source "${TEST_DIR}/launch_utils.sh" | |
| CONFIG_PATH="config/config_opd_7b_deepspeed.yaml" | |
| export ACCELERATE_CONFIG="${ACCELERATE_CONFIG:-$(resolve_deepspeed_zero1_config)}" | |
| prepare_fast_test_data "${CONFIG_PATH}" | |
| ensure_spacy_model | |
| NUM_PROCESSES="$(detect_num_gpus)" | |
| print_fast_plan "opd" "${CONFIG_PATH}" | |
| LOG_FILE="$(fast_train_log_path train_test_opd)" | |
| run_train_with_log "${LOG_FILE}" \ | |
| "${PYTHON_BIN}" -m accelerate.commands.launch --config_file "${ACCELERATE_CONFIG}" --num_processes "${NUM_PROCESSES}" main.py \ | |
| --config "${CONFIG_PATH}" \ | |
| --mode rl \ | |
| --opsd_enabled \ | |
| "$@" | |