NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

script test

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-19 16:07:26 -05:00
parent c80481a3cc
commit e1f9b926bb
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
example_trainer/run_gsm8k_lora_matrix.sh
View file

@ -30,6 +30,8 @@ set -euo pipefail
# MODE=all # one of: all, shared_vllm, lora_only, lora_restart
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
SCRIPT_PATH="${SCRIPT_DIR}/$(basename "${BASH_SOURCE[0]}")"
LAUNCH_DIR="$PWD"
cd "$ROOT_DIR"
@ -453,7 +455,7 @@ if [[ "$PARALLEL" == "1" ]]; then
for m in shared_vllm lora_only lora_restart; do
local_log="${OUTPUT_BASE_DIR}/logs/gsm8k_${m}/orchestrator.log"
printf ' '
printf '%q ' env MODE="$m" PARALLEL=0 "$0"
printf '%q ' env MODE="$m" PARALLEL=0 "$SCRIPT_PATH"
printf '> %q 2>&1 &\n' "$local_log"
done
log "[DRY RUN] parent waits for all child mode runners."
@ -466,7 +468,7 @@ if [[ "$PARALLEL" == "1" ]]; then
mkdir -p "$mode_log_dir"
mode_orch_log="${mode_log_dir}/orchestrator.log"
log "Starting mode runner: ${m} (log: ${mode_orch_log})"
env MODE="$m" PARALLEL=0 "$0" >"$mode_orch_log" 2>&1 &
env MODE="$m" PARALLEL=0 "$SCRIPT_PATH" >"$mode_orch_log" 2>&1 &
parallel_pids+=("$!")
done