Upload artifacts/training/scripts/run_pretrain_id2-10_0.25easy_0.25medium_0.5hard.sh

artifacts/training/scripts/run_pretrain_id2-10_0.25easy_0.25medium_0.5hard.sh

@@ -0,0 +1,62 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+REPO_ROOT="$(cd "${SCRIPT_DIR}/../../../.." && pwd)"
+cd "${REPO_ROOT}"
+
+# llamafactory-cli launches distributed training via `torchrun`, so the venv
+# bin dir must be on PATH even when the CLI itself is invoked by absolute path.
+export PATH="${REPO_ROOT}/.venv/bin:${PATH}"
+
+if [[ -n "${SLURM_CPUS_PER_TASK:-}" ]]; then
+    CPU_WORKERS_DEFAULT="${SLURM_CPUS_PER_TASK}"
+elif command -v nproc >/dev/null 2>&1; then
+    CPU_WORKERS_DEFAULT="$(nproc)"
+else
+    CPU_WORKERS_DEFAULT=1
+fi
+
+DEFAULT_PREPROCESSING_WORKERS="${PREPROCESSING_NUM_WORKERS:-${CPU_WORKERS_DEFAULT}}"
+DEFAULT_DATALOADER_WORKERS="${DATALOADER_NUM_WORKERS:-${CPU_WORKERS_DEFAULT}}"
+
+if [[ -z "${CUDA_VISIBLE_DEVICES:-}" ]]; then
+    echo "CUDA_VISIBLE_DEVICES must be set before running this script" >&2
+    exit 1
+fi
+
+LLAMA_BIN_DEFAULT="${REPO_ROOT}/.venv/bin/llamafactory-cli"
+DATASET_DIR_ROOT="${DATASET_DIR_ROOT:-data}"
+if [[ ! -x "${LLAMA_BIN_DEFAULT}" ]]; then
+    echo "Missing ${LLAMA_BIN_DEFAULT}. Run scripts/setup/install_local_llamafactory.sh first." >&2
+    exit 1
+fi
+
+if [[ ! -d "${REPO_ROOT}/${DATASET_DIR_ROOT}/composition/train" ]]; then
+    echo "Missing ${DATASET_DIR_ROOT}/composition/train. Run scripts/composition/prepare_hf_composition_data.sh first." >&2
+    exit 1
+fi
+
+if [[ ! -d "${REPO_ROOT}/${DATASET_DIR_ROOT}/composition/test" ]]; then
+    echo "Missing ${DATASET_DIR_ROOT}/composition/test. Run scripts/composition/prepare_hf_composition_data.sh first." >&2
+    exit 1
+fi
+
+export WANDB_PROJECT="${WANDB_PROJECT:-Interplay-LM-Reasoning}"
+export WANDB_ENTITY="${WANDB_ENTITY:-kmchiti}"
+
+DEFAULT_LLAMA_ARGS=(
+    "preprocessing_num_workers=${DEFAULT_PREPROCESSING_WORKERS}"
+    "dataloader_num_workers=${DEFAULT_DATALOADER_WORKERS}"
+)
+
+if [[ -n "${LLAMA_EXTRA_ARGS:-}" ]]; then
+    export LLAMA_EXTRA_ARGS="${DEFAULT_LLAMA_ARGS[*]} ${LLAMA_EXTRA_ARGS}"
+else
+    export LLAMA_EXTRA_ARGS="${DEFAULT_LLAMA_ARGS[*]}"
+fi
+
+EVAL_DATA_ROOT="${EVAL_DATA_ROOT:-${DATASET_DIR_ROOT}/composition/test}" \
+LLAMA_BIN="${LLAMA_BIN:-${LLAMA_BIN_DEFAULT}}" \
+LLAMA_CONFIG="scripts/composition/op-difficulty-10B/pt-diff2_10-tok10B-lr1e-4-bs512k-schedcos-minlr3e-5/id2-10_0.25easy_0.25medium_0.5hard.yaml" \
+    ./scripts/meta_run.sh --skip-rl "$@"