Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- example/timerpo_trainer/run_anomseer.sh +63 -0
- example/timerpo_trainer/run_anomseer_rats.sh +73 -0
- example/timerpo_trainer/run_rats_2gpu.sh +263 -0
- logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_053549.log +417 -0
- logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_122910.log +1196 -0
- logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_154822.log +0 -0
- logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_20260615.log +635 -0
- logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_v2_20260615.log +803 -0
- logs/rats_eval_chunked_check.log +0 -0
- logs/rats_eval_full.log +569 -0
- logs/rats_eval_full2.log +0 -0
- logs/rats_eval_full2_metrics.json +18 -0
- logs/rats_train_lora_20260611_174333.log +0 -0
- multimodal_data_processing/anom.py +596 -0
- multimodal_data_processing/rats_uni.py +268 -0
- outputs/2026-06-09/15-29-24/.hydra/config.yaml +194 -0
- outputs/2026-06-09/15-29-24/.hydra/hydra.yaml +206 -0
- outputs/2026-06-09/15-29-24/.hydra/overrides.yaml +52 -0
- outputs/2026-06-09/15-29-24/main_ppo.log +0 -0
- outputs/2026-06-09/16-19-04/.hydra/config.yaml +194 -0
- outputs/2026-06-09/16-19-04/.hydra/hydra.yaml +206 -0
- outputs/2026-06-09/16-19-04/.hydra/overrides.yaml +52 -0
- outputs/2026-06-09/16-19-04/main_ppo.log +0 -0
- outputs/2026-06-09/16-56-49/.hydra/config.yaml +194 -0
- outputs/2026-06-09/16-56-49/.hydra/hydra.yaml +206 -0
- outputs/2026-06-09/16-56-49/.hydra/overrides.yaml +52 -0
- outputs/2026-06-09/16-56-49/main_ppo.log +0 -0
- outputs/2026-06-09/17-08-35/.hydra/config.yaml +194 -0
- outputs/2026-06-09/17-08-35/.hydra/hydra.yaml +206 -0
- outputs/2026-06-09/17-08-35/.hydra/overrides.yaml +52 -0
- outputs/2026-06-09/17-08-35/main_ppo.log +0 -0
- outputs/2026-06-09/17-25-05/.hydra/config.yaml +194 -0
- outputs/2026-06-09/17-25-05/.hydra/hydra.yaml +206 -0
- outputs/2026-06-09/17-25-05/.hydra/overrides.yaml +52 -0
- outputs/2026-06-09/17-25-05/main_ppo.log +0 -0
- outputs/2026-06-09/17-44-32/.hydra/config.yaml +194 -0
- outputs/2026-06-09/17-44-32/.hydra/hydra.yaml +207 -0
- outputs/2026-06-09/17-44-32/.hydra/overrides.yaml +53 -0
- outputs/2026-06-09/17-44-32/main_ppo.log +0 -0
- outputs/2026-06-09/18-11-50/.hydra/config.yaml +194 -0
- outputs/2026-06-09/18-11-50/.hydra/hydra.yaml +210 -0
- outputs/2026-06-09/18-11-50/.hydra/overrides.yaml +56 -0
- outputs/2026-06-09/18-20-25/.hydra/config.yaml +194 -0
- outputs/2026-06-09/18-20-25/.hydra/overrides.yaml +53 -0
- outputs/2026-06-09/18-33-56/.hydra/hydra.yaml +216 -0
- outputs/2026-06-15/11-17-58/.hydra/hydra.yaml +210 -0
- outputs/2026-06-15/11-17-58/main_ppo.log +0 -0
- outputs/2026-06-15/11-25-36/.hydra/config.yaml +194 -0
- outputs/2026-06-15/11-25-36/.hydra/overrides.yaml +56 -0
- outputs/2026-06-15/11-25-36/main_ppo.log +0 -0
example/timerpo_trainer/run_anomseer.sh
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
|
| 3 |
+
MODEL_PATH=Qwen/Qwen2.5-VL-3B-Instruct
|
| 4 |
+
# MODEL_PATH=Qwen/Qwen2.5-VL-7B-Instruct
|
| 5 |
+
EVAL=False
|
| 6 |
+
|
| 7 |
+
TRAIN_FILE=${TRAIN_FILE:-./data/anol_processed_mllm_data/train_full.parquet}
|
| 8 |
+
VAL_FILE=${VAL_FILE:-./data/anol_processed_mllm_data/test_full.parquet}
|
| 9 |
+
|
| 10 |
+
if [ "$EVAL" = "True" ]; then
|
| 11 |
+
VAL_ONLY=True
|
| 12 |
+
else
|
| 13 |
+
VAL_ONLY=False
|
| 14 |
+
fi
|
| 15 |
+
|
| 16 |
+
python3 -m verl.trainer.main_ppo \
|
| 17 |
+
algorithm.adv_estimator=grpo \
|
| 18 |
+
data.train_files=$TRAIN_FILE \
|
| 19 |
+
data.val_files=$VAL_FILE \
|
| 20 |
+
data.train_batch_size=128 \
|
| 21 |
+
data.max_prompt_length=1024 \
|
| 22 |
+
data.max_response_length=1024 \
|
| 23 |
+
data.filter_overlong_prompts=True \
|
| 24 |
+
data.truncation='error' \
|
| 25 |
+
data.image_key=images \
|
| 26 |
+
actor_rollout_ref.model.path=$MODEL_PATH \
|
| 27 |
+
actor_rollout_ref.actor.optim.lr=1e-6 \
|
| 28 |
+
actor_rollout_ref.model.use_remove_padding=True \
|
| 29 |
+
actor_rollout_ref.actor.ppo_mini_batch_size=128 \
|
| 30 |
+
actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=8 \
|
| 31 |
+
actor_rollout_ref.actor.use_kl_loss=True \
|
| 32 |
+
actor_rollout_ref.actor.kl_loss_coef=0.001 \
|
| 33 |
+
actor_rollout_ref.actor.kl_loss_type=low_var_kl \
|
| 34 |
+
actor_rollout_ref.model.enable_gradient_checkpointing=True \
|
| 35 |
+
actor_rollout_ref.actor.fsdp_config.param_offload=False \
|
| 36 |
+
actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \
|
| 37 |
+
actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=16 \
|
| 38 |
+
actor_rollout_ref.rollout.tensor_model_parallel_size=2 \
|
| 39 |
+
actor_rollout_ref.rollout.name=vllm \
|
| 40 |
+
actor_rollout_ref.rollout.gpu_memory_utilization=0.6 \
|
| 41 |
+
actor_rollout_ref.rollout.enable_chunked_prefill=False \
|
| 42 |
+
actor_rollout_ref.rollout.enforce_eager=False \
|
| 43 |
+
actor_rollout_ref.rollout.free_cache_engine=False \
|
| 44 |
+
actor_rollout_ref.rollout.n=5 \
|
| 45 |
+
actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=16 \
|
| 46 |
+
actor_rollout_ref.ref.fsdp_config.param_offload=True \
|
| 47 |
+
algorithm.kl_ctrl.kl_coef=0.001 \
|
| 48 |
+
trainer.critic_warmup=0 \
|
| 49 |
+
trainer.logger=['console','wandb'] \
|
| 50 |
+
trainer.project_name='anomseer' \
|
| 51 |
+
trainer.experiment_name='anomseer_timerpo' \
|
| 52 |
+
trainer.n_gpus_per_node=4 \
|
| 53 |
+
trainer.nnodes=1 \
|
| 54 |
+
trainer.save_freq=500 \
|
| 55 |
+
trainer.test_freq=10 \
|
| 56 |
+
trainer.val_only=$VAL_ONLY \
|
| 57 |
+
trainer.val_before_train=True \
|
| 58 |
+
trainer.total_epochs=10 \
|
| 59 |
+
ts.use_sem_orth=True \
|
| 60 |
+
ts.adv_mix=0.3 \
|
| 61 |
+
ts.similarity_method=ot \
|
| 62 |
+
ts.ot_eps=0.08 \
|
| 63 |
+
ts.ot_n_iter=50 $@
|
example/timerpo_trainer/run_anomseer_rats.sh
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
# AnomSeer (TimerPO) on the Time-RA RATs-Uni univariate dataset.
|
| 3 |
+
# Task: 15-class anomaly classification + reasoning (no interval localization).
|
| 4 |
+
#
|
| 5 |
+
# Prepare data first:
|
| 6 |
+
# python multimodal_data_processing/rats_uni.py \
|
| 7 |
+
# --json_path /path/to/RATs40K/RATs-Uni-TSImage_Reason.json \
|
| 8 |
+
# --out_dir ./data/rats_uni_processed
|
| 9 |
+
#
|
| 10 |
+
# Train: bash example/timerpo_trainer/run_anomseer_rats.sh
|
| 11 |
+
# Eval (val only): EVAL=True bash example/timerpo_trainer/run_anomseer_rats.sh
|
| 12 |
+
|
| 13 |
+
MODEL_PATH=${MODEL_PATH:-Qwen/Qwen2.5-VL-3B-Instruct}
|
| 14 |
+
# MODEL_PATH=Qwen/Qwen2.5-VL-7B-Instruct
|
| 15 |
+
EVAL=${EVAL:-False}
|
| 16 |
+
|
| 17 |
+
TRAIN_FILE=${TRAIN_FILE:-./data/rats_uni_processed/train_full.parquet}
|
| 18 |
+
VAL_FILE=${VAL_FILE:-./data/rats_uni_processed/test_full.parquet}
|
| 19 |
+
|
| 20 |
+
if [ "$EVAL" = "True" ]; then
|
| 21 |
+
VAL_ONLY=True
|
| 22 |
+
else
|
| 23 |
+
VAL_ONLY=False
|
| 24 |
+
fi
|
| 25 |
+
|
| 26 |
+
python3 -m verl.trainer.main_ppo \
|
| 27 |
+
algorithm.adv_estimator=grpo \
|
| 28 |
+
data.train_files=$TRAIN_FILE \
|
| 29 |
+
data.val_files=$VAL_FILE \
|
| 30 |
+
data.train_batch_size=128 \
|
| 31 |
+
data.max_prompt_length=1024 \
|
| 32 |
+
data.max_response_length=1024 \
|
| 33 |
+
data.filter_overlong_prompts=True \
|
| 34 |
+
data.truncation='error' \
|
| 35 |
+
data.image_key=images \
|
| 36 |
+
actor_rollout_ref.model.path=$MODEL_PATH \
|
| 37 |
+
actor_rollout_ref.actor.optim.lr=1e-6 \
|
| 38 |
+
actor_rollout_ref.model.use_remove_padding=True \
|
| 39 |
+
actor_rollout_ref.actor.ppo_mini_batch_size=128 \
|
| 40 |
+
actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=8 \
|
| 41 |
+
actor_rollout_ref.actor.use_kl_loss=True \
|
| 42 |
+
actor_rollout_ref.actor.kl_loss_coef=0.001 \
|
| 43 |
+
actor_rollout_ref.actor.kl_loss_type=low_var_kl \
|
| 44 |
+
actor_rollout_ref.model.enable_gradient_checkpointing=True \
|
| 45 |
+
actor_rollout_ref.actor.fsdp_config.param_offload=False \
|
| 46 |
+
actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \
|
| 47 |
+
actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=16 \
|
| 48 |
+
actor_rollout_ref.rollout.tensor_model_parallel_size=2 \
|
| 49 |
+
actor_rollout_ref.rollout.name=vllm \
|
| 50 |
+
actor_rollout_ref.rollout.gpu_memory_utilization=0.6 \
|
| 51 |
+
actor_rollout_ref.rollout.enable_chunked_prefill=False \
|
| 52 |
+
actor_rollout_ref.rollout.enforce_eager=False \
|
| 53 |
+
actor_rollout_ref.rollout.free_cache_engine=False \
|
| 54 |
+
actor_rollout_ref.rollout.n=5 \
|
| 55 |
+
actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=16 \
|
| 56 |
+
actor_rollout_ref.ref.fsdp_config.param_offload=True \
|
| 57 |
+
algorithm.kl_ctrl.kl_coef=0.001 \
|
| 58 |
+
trainer.critic_warmup=0 \
|
| 59 |
+
trainer.logger=['console','wandb'] \
|
| 60 |
+
trainer.project_name='anomseer' \
|
| 61 |
+
trainer.experiment_name='anomseer_rats_uni' \
|
| 62 |
+
trainer.n_gpus_per_node=4 \
|
| 63 |
+
trainer.nnodes=1 \
|
| 64 |
+
trainer.save_freq=500 \
|
| 65 |
+
trainer.test_freq=10 \
|
| 66 |
+
trainer.val_only=$VAL_ONLY \
|
| 67 |
+
trainer.val_before_train=True \
|
| 68 |
+
trainer.total_epochs=10 \
|
| 69 |
+
ts.use_sem_orth=True \
|
| 70 |
+
ts.adv_mix=0.3 \
|
| 71 |
+
ts.similarity_method=ot \
|
| 72 |
+
ts.ot_eps=0.08 \
|
| 73 |
+
ts.ot_n_iter=50 $@
|
example/timerpo_trainer/run_rats_2gpu.sh
ADDED
|
@@ -0,0 +1,263 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
# =============================================================================
|
| 3 |
+
# AnomSeer / TimerPO on Time-RA RATs-Uni — 2-GPU train + eval launcher.
|
| 4 |
+
#
|
| 5 |
+
# Task : 15-class anomaly classification + reasoning (no localization).
|
| 6 |
+
# Tuning: LoRA fine-tuning (GRPO) by default.
|
| 7 |
+
#
|
| 8 |
+
# Modes (STAGE):
|
| 9 |
+
# train : train only (with periodic in-training validation) [default]
|
| 10 |
+
# eval : evaluate MODEL_PATH only (val_only)
|
| 11 |
+
# train_eval : train -> auto-merge latest checkpoint -> evaluate it
|
| 12 |
+
#
|
| 13 |
+
# Examples:
|
| 14 |
+
# bash example/timerpo_trainer/run_rats_2gpu.sh # train
|
| 15 |
+
# STAGE=eval MODEL_PATH=Qwen/Qwen2.5-VL-3B-Instruct \
|
| 16 |
+
# bash example/timerpo_trainer/run_rats_2gpu.sh # zero-shot eval
|
| 17 |
+
# STAGE=train_eval bash example/timerpo_trainer/run_rats_2gpu.sh # train then eval
|
| 18 |
+
# =============================================================================
|
| 19 |
+
set -euo pipefail
|
| 20 |
+
|
| 21 |
+
# ---- scratch on /dev/shm (NOT /tmp) ----------------------------------------
|
| 22 |
+
# The CIFS share (/mnt/share01) cannot host Ray's unix sockets, and the local root
|
| 23 |
+
# disk (/, where /tmp and ~/.cache live) is ~100% full. So Ray's runtime/spill and
|
| 24 |
+
# the torch/triton/vLLM/HF caches go to /dev/shm: local tmpfs, supports sockets,
|
| 25 |
+
# ~120 GB free. Persistent outputs (checkpoints, logs) still go under the project.
|
| 26 |
+
SCRATCH_ROOT="${SCRATCH_ROOT:-/dev/shm/anomseer}"
|
| 27 |
+
export RAY_TMPDIR="${RAY_TMPDIR:-${SCRATCH_ROOT}/ray}"
|
| 28 |
+
export TMPDIR="${TMPDIR:-${SCRATCH_ROOT}/tmp}"
|
| 29 |
+
export XDG_CACHE_HOME="${XDG_CACHE_HOME:-${SCRATCH_ROOT}/cache}"
|
| 30 |
+
export TORCHINDUCTOR_CACHE_DIR="${TORCHINDUCTOR_CACHE_DIR:-${SCRATCH_ROOT}/cache/torchinductor}"
|
| 31 |
+
export TRITON_CACHE_DIR="${TRITON_CACHE_DIR:-${SCRATCH_ROOT}/cache/triton}"
|
| 32 |
+
export VLLM_CACHE_ROOT="${VLLM_CACHE_ROOT:-${SCRATCH_ROOT}/cache/vllm}"
|
| 33 |
+
export HF_HOME="${HF_HOME:-${SCRATCH_ROOT}/cache/hf}"
|
| 34 |
+
mkdir -p "$RAY_TMPDIR" "$TMPDIR" "$XDG_CACHE_HOME" "$TORCHINDUCTOR_CACHE_DIR" \
|
| 35 |
+
"$TRITON_CACHE_DIR" "$VLLM_CACHE_ROOT" "$HF_HOME"
|
| 36 |
+
|
| 37 |
+
# ---- knobs (override via env) ----------------------------------------------
|
| 38 |
+
PYTHON_BIN="${PYTHON_BIN:-/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python}"
|
| 39 |
+
MODEL_PATH=${MODEL_PATH:-/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct} # local 3B; set to the 7B path / HF id to switch
|
| 40 |
+
STAGE=${STAGE:-train} # train | eval | train_eval
|
| 41 |
+
N_GPUS=${N_GPUS:-2} # number of GPUs (this script targets 2)
|
| 42 |
+
# TP = rollout tensor-parallel size; must divide N_GPUS. Default 2 splits vLLM's model
|
| 43 |
+
# across both cards (safer on memory). Set TP=1 for data-parallel rollout (full replica
|
| 44 |
+
# per card, no cross-card TP comm — faster on these no-NVLink cards; fine for 3B on 48 GB).
|
| 45 |
+
TP=${TP:-2}
|
| 46 |
+
GPU_MEM_UTIL=${GPU_MEM_UTIL:-0.4} # vllm KV-cache fraction (lower if OOM)
|
| 47 |
+
TRAIN_BATCH=${TRAIN_BATCH:-16} # prompts fetched per training step (rollout batch)
|
| 48 |
+
MICRO_BSZ=${MICRO_BSZ:-2} # actor ppo micro batch per GPU (lower if OOM)
|
| 49 |
+
LOGP_MICRO_BSZ=${LOGP_MICRO_BSZ:-8} # log-prob micro batch per GPU
|
| 50 |
+
PARAM_OFFLOAD=${PARAM_OFFLOAD:-False} # set True to offload actor params (saves VRAM)
|
| 51 |
+
OPTIM_OFFLOAD=${OPTIM_OFFLOAD:-False} # set True to offload optimizer (saves VRAM)
|
| 52 |
+
MAX_RESPONSE_LENGTH=${MAX_RESPONSE_LENGTH:-384} # covers the expert explanations without runaway output
|
| 53 |
+
KL_COEF=${KL_COEF:-0.01} # resist reward-driven mode collapse
|
| 54 |
+
# remove-padding (rmpad) is a throughput optimization that monkey-patches Qwen2VL
|
| 55 |
+
# FlashAttention2 internals. Those classes were removed in transformers>=4.52, so it
|
| 56 |
+
# must stay False on this env (transformers 4.54.1). Set True only on transformers<=4.51.1.
|
| 57 |
+
USE_RMPAD=${USE_RMPAD:-False}
|
| 58 |
+
EPOCHS=${EPOCHS:-1}
|
| 59 |
+
|
| 60 |
+
# ---- LoRA (optional) -------------------------------------------------------
|
| 61 |
+
# LORA_RANK=16 (default) -> LoRA fine-tuning.
|
| 62 |
+
# LORA_RANK=0 -> full-parameter fine-tuning.
|
| 63 |
+
LORA_RANK=${LORA_RANK:-16}
|
| 64 |
+
LORA_ALPHA=${LORA_ALPHA:-16}
|
| 65 |
+
LORA_DROPOUT=${LORA_DROPOUT:-0.0}
|
| 66 |
+
# NOTE: PEFT 'all-linear' tries to wrap whole Qwen2_5_VLVisionBlock modules and errors on
|
| 67 |
+
# this VL model, so default to the explicit LLM (+vision MLP) projection names instead.
|
| 68 |
+
LORA_TARGET_MODULES=${LORA_TARGET_MODULES:-q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj}
|
| 69 |
+
|
| 70 |
+
# LoRA usually wants a larger LR than full fine-tuning; default accordingly (override with LR=).
|
| 71 |
+
if [ "$LORA_RANK" -gt 0 ]; then
|
| 72 |
+
LR=${LR:-1e-5}
|
| 73 |
+
else
|
| 74 |
+
LR=${LR:-1e-6}
|
| 75 |
+
fi
|
| 76 |
+
LOGGER=${LOGGER:-console} # 'console' or "console','wandb" (needs wandb login)
|
| 77 |
+
|
| 78 |
+
PROJECT=${PROJECT:-anomseer}
|
| 79 |
+
EXP=${EXP:-anomseer_rats_uni_2gpu}
|
| 80 |
+
CKPT_ROOT=${CKPT_ROOT:-checkpoints/${PROJECT}/${EXP}}
|
| 81 |
+
|
| 82 |
+
# Default trains on a random 1/4 subset (7566 samples) for faster runs; override with
|
| 83 |
+
# TRAIN_FILE=./data/rats_uni_processed/train_full.parquet for the full 30266.
|
| 84 |
+
TRAIN_FILE=${TRAIN_FILE:-./data/rats_uni_processed/train_quarter.parquet}
|
| 85 |
+
# verl sends the WHOLE val set to vLLM in one batch, so the full 6034-sample test set
|
| 86 |
+
# OOMs system RAM. Use a stratified 474-sample subset for in-training validation; for a
|
| 87 |
+
# final full-test eval pass VAL_FILE=./data/rats_uni_processed/test_full.parquet.
|
| 88 |
+
VAL_FILE=${VAL_FILE:-./data/rats_uni_processed/test_small.parquet}
|
| 89 |
+
|
| 90 |
+
# ---- core launcher ---------------------------------------------------------
|
| 91 |
+
# args: $1 = val_only (True/False) $2 = model path
|
| 92 |
+
run_verl () {
|
| 93 |
+
local VAL_ONLY="$1"; local MPATH="$2"
|
| 94 |
+
local ACTIVE_LORA_RANK="$LORA_RANK"
|
| 95 |
+
local RESUME_MODE="auto"
|
| 96 |
+
if [ "$VAL_ONLY" = "True" ]; then
|
| 97 |
+
# Evaluation consumes a complete HF model, including any already-merged adapter.
|
| 98 |
+
ACTIVE_LORA_RANK=0
|
| 99 |
+
RESUME_MODE="disable"
|
| 100 |
+
fi
|
| 101 |
+
"$PYTHON_BIN" -m verl.trainer.main_ppo \
|
| 102 |
+
algorithm.adv_estimator=grpo \
|
| 103 |
+
data.train_files="$TRAIN_FILE" \
|
| 104 |
+
data.val_files="$VAL_FILE" \
|
| 105 |
+
data.train_batch_size="$TRAIN_BATCH" \
|
| 106 |
+
data.max_prompt_length=1024 \
|
| 107 |
+
data.max_response_length="$MAX_RESPONSE_LENGTH" \
|
| 108 |
+
data.filter_overlong_prompts=True \
|
| 109 |
+
data.truncation='error' \
|
| 110 |
+
data.image_key=images \
|
| 111 |
+
actor_rollout_ref.model.path="$MPATH" \
|
| 112 |
+
actor_rollout_ref.model.lora_rank="$ACTIVE_LORA_RANK" \
|
| 113 |
+
actor_rollout_ref.model.lora_alpha="$LORA_ALPHA" \
|
| 114 |
+
actor_rollout_ref.model.lora_dropout="$LORA_DROPOUT" \
|
| 115 |
+
actor_rollout_ref.model.lora_target_modules="'$LORA_TARGET_MODULES'" \
|
| 116 |
+
actor_rollout_ref.actor.optim.lr="$LR" \
|
| 117 |
+
actor_rollout_ref.model.use_remove_padding="$USE_RMPAD" \
|
| 118 |
+
actor_rollout_ref.actor.ppo_mini_batch_size="$TRAIN_BATCH" \
|
| 119 |
+
actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu="$MICRO_BSZ" \
|
| 120 |
+
actor_rollout_ref.actor.use_kl_loss=True \
|
| 121 |
+
actor_rollout_ref.actor.kl_loss_coef="$KL_COEF" \
|
| 122 |
+
actor_rollout_ref.actor.kl_loss_type=low_var_kl \
|
| 123 |
+
actor_rollout_ref.model.enable_gradient_checkpointing=True \
|
| 124 |
+
actor_rollout_ref.actor.fsdp_config.param_offload="$PARAM_OFFLOAD" \
|
| 125 |
+
actor_rollout_ref.actor.fsdp_config.optimizer_offload="$OPTIM_OFFLOAD" \
|
| 126 |
+
actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu="$LOGP_MICRO_BSZ" \
|
| 127 |
+
actor_rollout_ref.rollout.tensor_model_parallel_size="$TP" \
|
| 128 |
+
actor_rollout_ref.rollout.name=vllm \
|
| 129 |
+
actor_rollout_ref.rollout.gpu_memory_utilization="$GPU_MEM_UTIL" \
|
| 130 |
+
actor_rollout_ref.rollout.stop='["</class>"]' \
|
| 131 |
+
actor_rollout_ref.rollout.include_stop_str_in_output=True \
|
| 132 |
+
actor_rollout_ref.rollout.enable_chunked_prefill=False \
|
| 133 |
+
actor_rollout_ref.rollout.enforce_eager=False \
|
| 134 |
+
actor_rollout_ref.rollout.free_cache_engine=False \
|
| 135 |
+
actor_rollout_ref.rollout.n=5 \
|
| 136 |
+
actor_rollout_ref.rollout.val_kwargs.do_sample=False \
|
| 137 |
+
actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu="$LOGP_MICRO_BSZ" \
|
| 138 |
+
actor_rollout_ref.ref.fsdp_config.param_offload=True \
|
| 139 |
+
algorithm.kl_ctrl.kl_coef="$KL_COEF" \
|
| 140 |
+
trainer.critic_warmup=0 \
|
| 141 |
+
trainer.logger="['${LOGGER}']" \
|
| 142 |
+
trainer.project_name="$PROJECT" \
|
| 143 |
+
trainer.experiment_name="$EXP" \
|
| 144 |
+
trainer.default_local_dir="$CKPT_ROOT" \
|
| 145 |
+
trainer.n_gpus_per_node="$N_GPUS" \
|
| 146 |
+
trainer.nnodes=1 \
|
| 147 |
+
trainer.save_freq=500 \
|
| 148 |
+
trainer.test_freq=10 \
|
| 149 |
+
trainer.val_only="$VAL_ONLY" \
|
| 150 |
+
trainer.val_before_train=True \
|
| 151 |
+
trainer.resume_mode="$RESUME_MODE" \
|
| 152 |
+
trainer.total_epochs="$EPOCHS" \
|
| 153 |
+
ts.use_sem_orth=True \
|
| 154 |
+
ts.adv_mix=0.3 \
|
| 155 |
+
ts.similarity_method=ot \
|
| 156 |
+
ts.ot_eps=0.08 \
|
| 157 |
+
ts.ot_n_iter=50 "${@:3}"
|
| 158 |
+
}
|
| 159 |
+
|
| 160 |
+
merge_latest_ckpt () {
|
| 161 |
+
local latest
|
| 162 |
+
latest=$(ls -d "${CKPT_ROOT}"/global_step_* 2>/dev/null | sort -t_ -k3 -n | tail -1 || true)
|
| 163 |
+
if [ -z "$latest" ]; then
|
| 164 |
+
echo "[ERROR] no checkpoint found under ${CKPT_ROOT}" >&2; exit 1
|
| 165 |
+
fi
|
| 166 |
+
if has_hf_weights "${latest}/actor/huggingface"; then
|
| 167 |
+
echo "[merge] reusing existing HF weights: ${latest}/actor/huggingface" >&2
|
| 168 |
+
echo "${latest}/actor/huggingface"
|
| 169 |
+
return
|
| 170 |
+
fi
|
| 171 |
+
echo "[merge] latest checkpoint: ${latest}/actor" >&2
|
| 172 |
+
"$PYTHON_BIN" scripts/model_merger.py \
|
| 173 |
+
--local_dir "${latest}/actor" \
|
| 174 |
+
--lora-alpha "$LORA_ALPHA" >&2
|
| 175 |
+
echo "${latest}/actor/huggingface" # only the path goes to stdout (captured by caller)
|
| 176 |
+
}
|
| 177 |
+
|
| 178 |
+
has_hf_weights () {
|
| 179 |
+
local path="$1"
|
| 180 |
+
[ -f "${path}/model.safetensors" ] ||
|
| 181 |
+
[ -f "${path}/model.safetensors.index.json" ] ||
|
| 182 |
+
[ -f "${path}/pytorch_model.bin" ] ||
|
| 183 |
+
[ -f "${path}/pytorch_model.bin.index.json" ]
|
| 184 |
+
}
|
| 185 |
+
|
| 186 |
+
prepare_eval_model () {
|
| 187 |
+
local model_path="$1"
|
| 188 |
+
if has_hf_weights "$model_path"; then
|
| 189 |
+
echo "$model_path"
|
| 190 |
+
return
|
| 191 |
+
fi
|
| 192 |
+
|
| 193 |
+
local actor_dir=""
|
| 194 |
+
if [ "$(basename "$model_path")" = "huggingface" ]; then
|
| 195 |
+
actor_dir="$(dirname "$model_path")"
|
| 196 |
+
elif [ -d "${model_path}/huggingface" ]; then
|
| 197 |
+
actor_dir="$model_path"
|
| 198 |
+
fi
|
| 199 |
+
|
| 200 |
+
if [ -z "$actor_dir" ] ||
|
| 201 |
+
! find "$actor_dir" -maxdepth 1 -name 'model_world_size_*_rank_0.pt' -print -quit | grep -q .; then
|
| 202 |
+
echo "[ERROR] '${model_path}' has no HF model weights and is not a mergeable FSDP checkpoint." >&2
|
| 203 |
+
return 1
|
| 204 |
+
fi
|
| 205 |
+
|
| 206 |
+
echo "[merge] HF weights missing; merging checkpoint ${actor_dir}" >&2
|
| 207 |
+
"$PYTHON_BIN" scripts/model_merger.py \
|
| 208 |
+
--local_dir "$actor_dir" \
|
| 209 |
+
--lora-alpha "$LORA_ALPHA" >&2
|
| 210 |
+
|
| 211 |
+
model_path="${actor_dir}/huggingface"
|
| 212 |
+
if ! has_hf_weights "$model_path"; then
|
| 213 |
+
echo "[ERROR] merge completed without producing HF model weights under '${model_path}'." >&2
|
| 214 |
+
return 1
|
| 215 |
+
fi
|
| 216 |
+
echo "$model_path"
|
| 217 |
+
}
|
| 218 |
+
|
| 219 |
+
# ---- log file --------------------------------------------------------------
|
| 220 |
+
# Tee all output (terminal + file). Override path with LOG_FILE=, or LOG_DIR=.
|
| 221 |
+
LOG_DIR=${LOG_DIR:-/mnt/share01/sqk/AnomSeer/logs}
|
| 222 |
+
mkdir -p "$LOG_DIR"
|
| 223 |
+
LOG_FILE=${LOG_FILE:-${LOG_DIR}/rats_${STAGE}$([ "$LORA_RANK" -gt 0 ] && echo _lora)_$(date +%Y%m%d_%H%M%S).log}
|
| 224 |
+
exec > >(tee -a "$LOG_FILE") 2>&1
|
| 225 |
+
echo "[log] saving full output to: $LOG_FILE"
|
| 226 |
+
|
| 227 |
+
# ---- dispatch --------------------------------------------------------------
|
| 228 |
+
if [ "$LORA_RANK" -gt 0 ]; then
|
| 229 |
+
FT_MODE="LoRA (r=${LORA_RANK}, alpha=${LORA_ALPHA}, target=${LORA_TARGET_MODULES})"
|
| 230 |
+
else
|
| 231 |
+
FT_MODE="full-parameter fine-tuning"
|
| 232 |
+
fi
|
| 233 |
+
if [ "$TP" -le 1 ]; then PARALLEL="data-parallel (DP=${N_GPUS}, full replica per card)"; else PARALLEL="tensor-parallel (TP=${TP})"; fi
|
| 234 |
+
echo "[config] tuning=${FT_MODE} | lr=${LR} | gpus=${N_GPUS} | ${PARALLEL}"
|
| 235 |
+
echo "[config] model=${MODEL_PATH}"
|
| 236 |
+
echo "[config] python=${PYTHON_BIN} | stage=${STAGE}"
|
| 237 |
+
if [ "$LORA_RANK" -eq 0 ] && [ "$PARAM_OFFLOAD" != "True" ] && [[ "$MODEL_PATH" == *7[bB]* ]]; then
|
| 238 |
+
echo "[hint] full-parameter 7B on 2x48GB will likely OOM. Either:"
|
| 239 |
+
echo " (a) LoRA: LORA_RANK=16 bash $0"
|
| 240 |
+
echo " (b) offload: PARAM_OFFLOAD=True OPTIM_OFFLOAD=True bash $0"
|
| 241 |
+
fi
|
| 242 |
+
|
| 243 |
+
case "$STAGE" in
|
| 244 |
+
train)
|
| 245 |
+
echo "[stage] TRAIN (model=${MODEL_PATH}, gpus=${N_GPUS}, tp=${TP})"
|
| 246 |
+
run_verl False "$MODEL_PATH" "$@"
|
| 247 |
+
;;
|
| 248 |
+
eval)
|
| 249 |
+
eval_model=$(prepare_eval_model "$MODEL_PATH")
|
| 250 |
+
echo "[stage] EVAL (model=${eval_model}, gpus=${N_GPUS}, tp=${TP}, lora_rank=0)"
|
| 251 |
+
run_verl True "$eval_model" "$@"
|
| 252 |
+
;;
|
| 253 |
+
train_eval)
|
| 254 |
+
echo "[stage] TRAIN (model=${MODEL_PATH}, gpus=${N_GPUS}, tp=${TP})"
|
| 255 |
+
run_verl False "$MODEL_PATH"
|
| 256 |
+
merged=$(merge_latest_ckpt)
|
| 257 |
+
echo "[stage] EVAL (merged checkpoint=${merged})"
|
| 258 |
+
run_verl True "$merged" "$@"
|
| 259 |
+
;;
|
| 260 |
+
*)
|
| 261 |
+
echo "[ERROR] unknown STAGE='$STAGE' (use train | eval | train_eval)" >&2; exit 1
|
| 262 |
+
;;
|
| 263 |
+
esac
|
logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_053549.log
ADDED
|
@@ -0,0 +1,417 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[log] saving full output to: /mnt/share01/sqk/AnomSeer/logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_053549.log
|
| 2 |
+
[config] tuning=LoRA (r=16, alpha=16, target=q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj) | lr=1e-4 | gpus=2 | tensor-parallel (TP=2)
|
| 3 |
+
[config] model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 4 |
+
[config] python=/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python | stage=eval
|
| 5 |
+
[stage] EVAL (model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface, gpus=2, tp=2)
|
| 6 |
+
2026-06-15 05:35:59,961 INFO worker.py:2012 -- Started a local Ray instance.
|
| 7 |
+
/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py:2051: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
|
| 8 |
+
warnings.warn(
|
| 9 |
+
[36m(TaskRunner pid=363397)[0m {'actor_rollout_ref': {'actor': {'clip_ratio': 0.2,
|
| 10 |
+
[36m(TaskRunner pid=363397)[0m 'entropy_coeff': 0.001,
|
| 11 |
+
[36m(TaskRunner pid=363397)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 12 |
+
[36m(TaskRunner pid=363397)[0m 'optimizer_offload': False,
|
| 13 |
+
[36m(TaskRunner pid=363397)[0m 'param_offload': False,
|
| 14 |
+
[36m(TaskRunner pid=363397)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 15 |
+
[36m(TaskRunner pid=363397)[0m 'grad_clip': 1.0,
|
| 16 |
+
[36m(TaskRunner pid=363397)[0m 'kl_loss_coef': 0.001,
|
| 17 |
+
[36m(TaskRunner pid=363397)[0m 'kl_loss_type': 'low_var_kl',
|
| 18 |
+
[36m(TaskRunner pid=363397)[0m 'optim': {'lr': 0.0001,
|
| 19 |
+
[36m(TaskRunner pid=363397)[0m 'lr_warmup_steps': -1,
|
| 20 |
+
[36m(TaskRunner pid=363397)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 21 |
+
[36m(TaskRunner pid=363397)[0m 'min_lr_ratio': None,
|
| 22 |
+
[36m(TaskRunner pid=363397)[0m 'total_training_steps': -1,
|
| 23 |
+
[36m(TaskRunner pid=363397)[0m 'warmup_style': 'constant'},
|
| 24 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_epochs': 1,
|
| 25 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_max_token_len_per_gpu': 16384,
|
| 26 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_micro_batch_size': None,
|
| 27 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_micro_batch_size_per_gpu': 2,
|
| 28 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_mini_batch_size': 16,
|
| 29 |
+
[36m(TaskRunner pid=363397)[0m 'shuffle': False,
|
| 30 |
+
[36m(TaskRunner pid=363397)[0m 'strategy': 'fsdp',
|
| 31 |
+
[36m(TaskRunner pid=363397)[0m 'ulysses_sequence_parallel_size': 1,
|
| 32 |
+
[36m(TaskRunner pid=363397)[0m 'use_dynamic_bsz': False,
|
| 33 |
+
[36m(TaskRunner pid=363397)[0m 'use_kl_loss': True,
|
| 34 |
+
[36m(TaskRunner pid=363397)[0m 'use_torch_compile': True},
|
| 35 |
+
[36m(TaskRunner pid=363397)[0m 'hybrid_engine': True,
|
| 36 |
+
[36m(TaskRunner pid=363397)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 37 |
+
[36m(TaskRunner pid=363397)[0m 'external_lib': None,
|
| 38 |
+
[36m(TaskRunner pid=363397)[0m 'lora_alpha': 16,
|
| 39 |
+
[36m(TaskRunner pid=363397)[0m 'lora_dropout': 0.0,
|
| 40 |
+
[36m(TaskRunner pid=363397)[0m 'lora_rank': 16,
|
| 41 |
+
[36m(TaskRunner pid=363397)[0m 'lora_target_modules': 'q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',
|
| 42 |
+
[36m(TaskRunner pid=363397)[0m 'override_config': {},
|
| 43 |
+
[36m(TaskRunner pid=363397)[0m 'path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 44 |
+
[36m(TaskRunner pid=363397)[0m 'use_remove_padding': False},
|
| 45 |
+
[36m(TaskRunner pid=363397)[0m 'ref': {'fsdp_config': {'param_offload': True,
|
| 46 |
+
[36m(TaskRunner pid=363397)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 47 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 48 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_micro_batch_size': None,
|
| 49 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 50 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_use_dynamic_bsz': False,
|
| 51 |
+
[36m(TaskRunner pid=363397)[0m 'ulysses_sequence_parallel_size': 1},
|
| 52 |
+
[36m(TaskRunner pid=363397)[0m 'rollout': {'disable_log_stats': True,
|
| 53 |
+
[36m(TaskRunner pid=363397)[0m 'do_sample': True,
|
| 54 |
+
[36m(TaskRunner pid=363397)[0m 'dtype': 'bfloat16',
|
| 55 |
+
[36m(TaskRunner pid=363397)[0m 'enable_chunked_prefill': False,
|
| 56 |
+
[36m(TaskRunner pid=363397)[0m 'enforce_eager': False,
|
| 57 |
+
[36m(TaskRunner pid=363397)[0m 'free_cache_engine': False,
|
| 58 |
+
[36m(TaskRunner pid=363397)[0m 'gpu_memory_utilization': 0.4,
|
| 59 |
+
[36m(TaskRunner pid=363397)[0m 'ignore_eos': False,
|
| 60 |
+
[36m(TaskRunner pid=363397)[0m 'load_format': 'dummy_dtensor',
|
| 61 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 62 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_micro_batch_size': None,
|
| 63 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 64 |
+
[36m(TaskRunner pid=363397)[0m 'log_prob_use_dynamic_bsz': False,
|
| 65 |
+
[36m(TaskRunner pid=363397)[0m 'max_model_len': None,
|
| 66 |
+
[36m(TaskRunner pid=363397)[0m 'max_num_batched_tokens': 8192,
|
| 67 |
+
[36m(TaskRunner pid=363397)[0m 'max_num_seqs': 1024,
|
| 68 |
+
[36m(TaskRunner pid=363397)[0m 'n': 5,
|
| 69 |
+
[36m(TaskRunner pid=363397)[0m 'name': 'vllm',
|
| 70 |
+
[36m(TaskRunner pid=363397)[0m 'prompt_length': 1024,
|
| 71 |
+
[36m(TaskRunner pid=363397)[0m 'response_length': 1024,
|
| 72 |
+
[36m(TaskRunner pid=363397)[0m 'temperature': 1.0,
|
| 73 |
+
[36m(TaskRunner pid=363397)[0m 'tensor_model_parallel_size': 2,
|
| 74 |
+
[36m(TaskRunner pid=363397)[0m 'top_k': -1,
|
| 75 |
+
[36m(TaskRunner pid=363397)[0m 'top_p': 1,
|
| 76 |
+
[36m(TaskRunner pid=363397)[0m 'use_fire_sampling': False,
|
| 77 |
+
[36m(TaskRunner pid=363397)[0m 'val_kwargs': {'do_sample': False,
|
| 78 |
+
[36m(TaskRunner pid=363397)[0m 'n': 1,
|
| 79 |
+
[36m(TaskRunner pid=363397)[0m 'temperature': 0.4,
|
| 80 |
+
[36m(TaskRunner pid=363397)[0m 'top_k': -1,
|
| 81 |
+
[36m(TaskRunner pid=363397)[0m 'top_p': 1.0}}},
|
| 82 |
+
[36m(TaskRunner pid=363397)[0m 'algorithm': {'adv_estimator': 'grpo',
|
| 83 |
+
[36m(TaskRunner pid=363397)[0m 'gamma': 1.0,
|
| 84 |
+
[36m(TaskRunner pid=363397)[0m 'kl_ctrl': {'kl_coef': 0.001, 'type': 'fixed'},
|
| 85 |
+
[36m(TaskRunner pid=363397)[0m 'kl_penalty': 'kl',
|
| 86 |
+
[36m(TaskRunner pid=363397)[0m 'lam': 1.0},
|
| 87 |
+
[36m(TaskRunner pid=363397)[0m 'critic': {'cliprange_value': 0.5,
|
| 88 |
+
[36m(TaskRunner pid=363397)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 89 |
+
[36m(TaskRunner pid=363397)[0m 'forward_micro_batch_size': None,
|
| 90 |
+
[36m(TaskRunner pid=363397)[0m 'forward_micro_batch_size_per_gpu': None,
|
| 91 |
+
[36m(TaskRunner pid=363397)[0m 'grad_clip': 1.0,
|
| 92 |
+
[36m(TaskRunner pid=363397)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 93 |
+
[36m(TaskRunner pid=363397)[0m 'external_lib': None,
|
| 94 |
+
[36m(TaskRunner pid=363397)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 95 |
+
[36m(TaskRunner pid=363397)[0m 'optimizer_offload': False,[36m(TaskRunner pid=363397)[0m DeprecationWarning: `ray.state.available_resources_per_node` is a private attribute and access will be removed in a future Ray version.
|
| 96 |
+
|
| 97 |
+
[36m(TaskRunner pid=363397)[0m 'param_offload': False,
|
| 98 |
+
[36m(TaskRunner pid=363397)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 99 |
+
[36m(TaskRunner pid=363397)[0m 'override_config': {},
|
| 100 |
+
[36m(TaskRunner pid=363397)[0m 'path': '~/models/deepseek-llm-7b-chat',
|
| 101 |
+
[36m(TaskRunner pid=363397)[0m 'tokenizer_path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 102 |
+
[36m(TaskRunner pid=363397)[0m 'use_remove_padding': False},
|
| 103 |
+
[36m(TaskRunner pid=363397)[0m 'optim': {'lr': 1e-05,
|
| 104 |
+
[36m(TaskRunner pid=363397)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 105 |
+
[36m(TaskRunner pid=363397)[0m 'min_lr_ratio': None,
|
| 106 |
+
[36m(TaskRunner pid=363397)[0m 'total_training_steps': -1,
|
| 107 |
+
[36m(TaskRunner pid=363397)[0m 'warmup_style': 'constant'},
|
| 108 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_epochs': 1,
|
| 109 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_max_token_len_per_gpu': 32768,
|
| 110 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_micro_batch_size': None,
|
| 111 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_micro_batch_size_per_gpu': None,
|
| 112 |
+
[36m(TaskRunner pid=363397)[0m 'ppo_mini_batch_size': 16,
|
| 113 |
+
[36m(TaskRunner pid=363397)[0m 'shuffle': False,
|
| 114 |
+
[36m(TaskRunner pid=363397)[0m 'strategy': 'fsdp',
|
| 115 |
+
[36m(TaskRunner pid=363397)[0m 'ulysses_sequence_parallel_size': 1,
|
| 116 |
+
[36m(TaskRunner pid=363397)[0m 'use_dynamic_bsz': False},
|
| 117 |
+
[36m(TaskRunner pid=363397)[0m 'custom_reward_function': {'name': 'compute_score', 'path': None},
|
| 118 |
+
[36m(TaskRunner pid=363397)[0m 'data': {'filter_overlong_prompts': True,
|
| 119 |
+
[36m(TaskRunner pid=363397)[0m 'image_key': 'images',
|
| 120 |
+
[36m(TaskRunner pid=363397)[0m 'max_prompt_length': 1024,
|
| 121 |
+
[36m(TaskRunner pid=363397)[0m 'max_response_length': 1024,
|
| 122 |
+
[36m(TaskRunner pid=363397)[0m 'prompt_key': 'prompt',
|
| 123 |
+
[36m(TaskRunner pid=363397)[0m 'return_raw_chat': False,
|
| 124 |
+
[36m(TaskRunner pid=363397)[0m 'return_raw_input_ids': False,
|
| 125 |
+
[36m(TaskRunner pid=363397)[0m 'shuffle': True,
|
| 126 |
+
[36m(TaskRunner pid=363397)[0m 'tokenizer': None,
|
| 127 |
+
[36m(TaskRunner pid=363397)[0m 'train_batch_size': 16,
|
| 128 |
+
[36m(TaskRunner pid=363397)[0m 'train_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_full.parquet',
|
| 129 |
+
[36m(TaskRunner pid=363397)[0m 'truncation': 'error',
|
| 130 |
+
[36m(TaskRunner pid=363397)[0m 'val_batch_size': 64,
|
| 131 |
+
[36m(TaskRunner pid=363397)[0m 'val_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/test_full.parquet'},
|
| 132 |
+
[36m(TaskRunner pid=363397)[0m 'reward_model': {'enable': False,
|
| 133 |
+
[36m(TaskRunner pid=363397)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 134 |
+
[36m(TaskRunner pid=363397)[0m 'max_length': None,
|
| 135 |
+
[36m(TaskRunner pid=363397)[0m 'micro_batch_size': None,
|
| 136 |
+
[36m(TaskRunner pid=363397)[0m 'micro_batch_size_per_gpu': None,
|
| 137 |
+
[36m(TaskRunner pid=363397)[0m 'model': {'external_lib': None,
|
| 138 |
+
[36m(TaskRunner pid=363397)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 139 |
+
[36m(TaskRunner pid=363397)[0m 'param_offload': False,
|
| 140 |
+
[36m(TaskRunner pid=363397)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 141 |
+
[36m(TaskRunner pid=363397)[0m 'input_tokenizer': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 142 |
+
[36m(TaskRunner pid=363397)[0m 'path': '~/models/FsfairX-LLaMA3-RM-v0.1',
|
| 143 |
+
[36m(TaskRunner pid=363397)[0m 'use_remove_padding': False},
|
| 144 |
+
[36m(TaskRunner pid=363397)[0m 'reward_manager': 'naive',
|
| 145 |
+
[36m(TaskRunner pid=363397)[0m 'strategy': 'fsdp',
|
| 146 |
+
[36m(TaskRunner pid=363397)[0m 'ulysses_sequence_parallel_size': 1,
|
| 147 |
+
[36m(TaskRunner pid=363397)[0m 'use_dynamic_bsz': False},
|
| 148 |
+
[36m(TaskRunner pid=363397)[0m 'trainer': {'balance_batch': True,
|
| 149 |
+
[36m(TaskRunner pid=363397)[0m 'critic_warmup': 0,
|
| 150 |
+
[36m(TaskRunner pid=363397)[0m 'default_hdfs_dir': None,
|
| 151 |
+
[36m(TaskRunner pid=363397)[0m 'default_local_dir': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3',
|
| 152 |
+
[36m(TaskRunner pid=363397)[0m 'del_local_ckpt_after_load': False,
|
| 153 |
+
[36m(TaskRunner pid=363397)[0m 'experiment_name': 'anomseer_rats_uni_full_20260612_184427_exp3_full_eval',
|
| 154 |
+
[36m(TaskRunner pid=363397)[0m 'logger': ['console'],
|
| 155 |
+
[36m(TaskRunner pid=363397)[0m 'n_gpus_per_node': 2,
|
| 156 |
+
[36m(TaskRunner pid=363397)[0m 'nnodes': 1,
|
| 157 |
+
[36m(TaskRunner pid=363397)[0m 'project_name': 'anomseer',
|
| 158 |
+
[36m(TaskRunner pid=363397)[0m 'remove_previous_ckpt_in_save': False,
|
| 159 |
+
[36m(TaskRunner pid=363397)[0m 'resume_from_path': False,
|
| 160 |
+
[36m(TaskRunner pid=363397)[0m 'resume_mode': 'auto',
|
| 161 |
+
[36m(TaskRunner pid=363397)[0m 'save_freq': 500,
|
| 162 |
+
[36m(TaskRunner pid=363397)[0m 'test_freq': -1,
|
| 163 |
+
[36m(TaskRunner pid=363397)[0m 'total_epochs': 1,
|
| 164 |
+
[36m(TaskRunner pid=363397)[0m 'total_training_steps': None,
|
| 165 |
+
[36m(TaskRunner pid=363397)[0m 'val_before_train': True,
|
| 166 |
+
[36m(TaskRunner pid=363397)[0m 'val_generations_to_log_to_wandb': 0,
|
| 167 |
+
[36m(TaskRunner pid=363397)[0m 'val_only': True},
|
| 168 |
+
[36m(TaskRunner pid=363397)[0m 'ts': {'adv_mix': 0.3,
|
| 169 |
+
[36m(TaskRunner pid=363397)[0m 'clip_pooling': 'mean',
|
| 170 |
+
[36m(TaskRunner pid=363397)[0m 'clip_temperature': 1.0,
|
| 171 |
+
[36m(TaskRunner pid=363397)[0m 'ot_eps': 0.08,
|
| 172 |
+
[36m(TaskRunner pid=363397)[0m 'ot_n_iter': 50,
|
| 173 |
+
[36m(TaskRunner pid=363397)[0m 'similarity_method': 'ot',
|
| 174 |
+
[36m(TaskRunner pid=363397)[0m 'use_sem_orth': True}}
|
| 175 |
+
[36m(TaskRunner pid=363397)[0m reward_manager_name: anomts
|
| 176 |
+
[36m(TaskRunner pid=363397)[0m WARNING: val_batch_size is deprecated. Validation datasets are sent to inference engines as a whole batch, which will schedule the memory themselves.
|
| 177 |
+
[36m(TaskRunner pid=363397)[0m [validate_config] All configuration checks passed successfully!
|
| 178 |
+
[36m(TaskRunner pid=363397)[0m dataset len: 30266
|
| 179 |
+
[36m(TaskRunner pid=363397)[0m filter dataset len: 30266
|
| 180 |
+
[36m(TaskRunner pid=363397)[0m dataset len: 6034
|
| 181 |
+
[36m(TaskRunner pid=363397)[0m filter dataset len: 6034
|
| 182 |
+
[36m(TaskRunner pid=363397)[0m Size of train dataloader: 1891
|
| 183 |
+
[36m(TaskRunner pid=363397)[0m Total training steps: 1891
|
| 184 |
+
[36m(WorkerDict pid=364993)[0m Model config after override: Qwen2_5_VLConfig {
|
| 185 |
+
[36m(WorkerDict pid=364993)[0m "architectures": [
|
| 186 |
+
[36m(WorkerDict pid=364993)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 187 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 188 |
+
[36m(WorkerDict pid=364993)[0m "attention_dropout": 0.0,
|
| 189 |
+
[36m(WorkerDict pid=364993)[0m "eos_token_id": 151645,
|
| 190 |
+
[36m(WorkerDict pid=364993)[0m "hidden_act": "silu",
|
| 191 |
+
[36m(WorkerDict pid=364993)[0m "hidden_size": 2048,
|
| 192 |
+
[36m(WorkerDict pid=364993)[0m "image_token_id": 151655,
|
| 193 |
+
[36m(WorkerDict pid=364993)[0m "initializer_range": 0.02,
|
| 194 |
+
[36m(WorkerDict pid=364993)[0m "intermediate_size": 11008,
|
| 195 |
+
[36m(WorkerDict pid=364993)[0m "max_position_embeddings": 128000,
|
| 196 |
+
[36m(WorkerDict pid=364993)[0m "max_window_layers": 70,
|
| 197 |
+
[36m(WorkerDict pid=364993)[0m "model_type": "qwen2_5_vl",
|
| 198 |
+
[36m(WorkerDict pid=364993)[0m "num_attention_heads": 16,
|
| 199 |
+
[36m(WorkerDict pid=364993)[0m "num_hidden_layers": 36,
|
| 200 |
+
[36m(WorkerDict pid=364993)[0m "num_key_value_heads": 2,
|
| 201 |
+
[36m(WorkerDict pid=364993)[0m "pad_token_id": 151643,
|
| 202 |
+
[36m(WorkerDict pid=364993)[0m "rms_norm_eps": 1e-06,
|
| 203 |
+
[36m(WorkerDict pid=364993)[0m "rope_scaling": {
|
| 204 |
+
[36m(WorkerDict pid=364993)[0m "mrope_section": [
|
| 205 |
+
[36m(WorkerDict pid=364993)[0m 16,
|
| 206 |
+
[36m(WorkerDict pid=364993)[0m 24,
|
| 207 |
+
[36m(WorkerDict pid=364993)[0m 24
|
| 208 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 209 |
+
[36m(WorkerDict pid=364993)[0m "rope_type": "default",
|
| 210 |
+
[36m(WorkerDict pid=364993)[0m "type": "default"
|
| 211 |
+
[36m(WorkerDict pid=364993)[0m },
|
| 212 |
+
[36m(WorkerDict pid=364993)[0m "rope_theta": 1000000.0,
|
| 213 |
+
[36m(WorkerDict pid=364993)[0m "sliding_window": 32768,
|
| 214 |
+
[36m(WorkerDict pid=364993)[0m "text_config": {
|
| 215 |
+
[36m(WorkerDict pid=364993)[0m "architectures": [
|
| 216 |
+
[36m(WorkerDict pid=364993)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 217 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 218 |
+
[36m(WorkerDict pid=364993)[0m "attention_dropout": 0.0,
|
| 219 |
+
[36m(WorkerDict pid=364993)[0m "bos_token_id": 151643,
|
| 220 |
+
[36m(WorkerDict pid=364993)[0m "eos_token_id": 151645,
|
| 221 |
+
[36m(WorkerDict pid=364993)[0m "hidden_act": "silu",
|
| 222 |
+
[36m(WorkerDict pid=364993)[0m "hidden_size": 2048,
|
| 223 |
+
[36m(WorkerDict pid=364993)[0m "image_token_id": null,
|
| 224 |
+
[36m(WorkerDict pid=364993)[0m "initializer_range": 0.02,
|
| 225 |
+
[36m(WorkerDict pid=364993)[0m "intermediate_size": 11008,
|
| 226 |
+
[36m(WorkerDict pid=364993)[0m "layer_types": [
|
| 227 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 228 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 229 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 230 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 231 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 232 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 233 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 234 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 235 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 236 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 237 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 238 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 239 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 240 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 241 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 242 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 243 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 244 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 245 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 246 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 247 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 248 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 249 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 250 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 251 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 252 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 253 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 254 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 255 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 256 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 257 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 258 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 259 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 260 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 261 |
+
[36m(WorkerDict pid=364993)[0m "full_attention",
|
| 262 |
+
[36m(WorkerDict pid=364993)[0m "full_attention"
|
| 263 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 264 |
+
[36m(WorkerDict pid=364993)[0m "max_position_embeddings": 128000,
|
| 265 |
+
[36m(WorkerDict pid=364993)[0m "max_window_layers": 70,
|
| 266 |
+
[36m(WorkerDict pid=364993)[0m "model_type": "qwen2_5_vl_text",
|
| 267 |
+
[36m(WorkerDict pid=364993)[0m "num_attention_heads": 16,
|
| 268 |
+
[36m(WorkerDict pid=364993)[0m "num_hidden_layers": 36,
|
| 269 |
+
[36m(WorkerDict pid=364993)[0m "num_key_value_heads": 2,
|
| 270 |
+
[36m(WorkerDict pid=364993)[0m "rms_norm_eps": 1e-06,
|
| 271 |
+
[36m(WorkerDict pid=364993)[0m "rope_scaling": {
|
| 272 |
+
[36m(WorkerDict pid=364993)[0m "mrope_section": [
|
| 273 |
+
[36m(WorkerDict pid=364993)[0m 16,
|
| 274 |
+
[36m(WorkerDict pid=364993)[0m 24,
|
| 275 |
+
[36m(WorkerDict pid=364993)[0m 24
|
| 276 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 277 |
+
[36m(WorkerDict pid=364993)[0m "rope_type": "default",
|
| 278 |
+
[36m(WorkerDict pid=364993)[0m "type": "default"
|
| 279 |
+
[36m(WorkerDict pid=364993)[0m },
|
| 280 |
+
[36m(WorkerDict pid=364993)[0m "rope_theta": 1000000.0,
|
| 281 |
+
[36m(WorkerDict pid=364993)[0m "sliding_window": null,
|
| 282 |
+
[36m(WorkerDict pid=364993)[0m "tie_word_embeddings": true,
|
| 283 |
+
[36m(WorkerDict pid=364993)[0m "torch_dtype": "float32",
|
| 284 |
+
[36m(WorkerDict pid=364993)[0m "use_cache": true,
|
| 285 |
+
[36m(WorkerDict pid=364993)[0m "use_sliding_window": false,
|
| 286 |
+
[36m(WorkerDict pid=364993)[0m "video_token_id": null,
|
| 287 |
+
[36m(WorkerDict pid=364993)[0m "vision_end_token_id": 151653,
|
| 288 |
+
[36m(WorkerDict pid=364993)[0m "vision_start_token_id": 151652,
|
| 289 |
+
[36m(WorkerDict pid=364993)[0m "vision_token_id": 151654,
|
| 290 |
+
[36m(WorkerDict pid=364993)[0m "vocab_size": 151936
|
| 291 |
+
[36m(WorkerDict pid=364993)[0m },
|
| 292 |
+
[36m(WorkerDict pid=364993)[0m "torch_dtype": "float32",
|
| 293 |
+
[36m(WorkerDict pid=364993)[0m "transformers_version": "4.54.1",
|
| 294 |
+
[36m(WorkerDict pid=364993)[0m "use_cache": true,
|
| 295 |
+
[36m(WorkerDict pid=364993)[0m "use_sliding_window": false,
|
| 296 |
+
[36m(WorkerDict pid=364993)[0m "video_token_id": 151656,
|
| 297 |
+
[36m(WorkerDict pid=364993)[0m "vision_config": {
|
| 298 |
+
[36m(WorkerDict pid=364993)[0m "depth": 32,
|
| 299 |
+
[36m(WorkerDict pid=364993)[0m "fullatt_block_indexes": [
|
| 300 |
+
[36m(WorkerDict pid=364993)[0m 7,
|
| 301 |
+
[36m(WorkerDict pid=364993)[0m 15,
|
| 302 |
+
[36m(WorkerDict pid=364993)[0m 23,
|
| 303 |
+
[36m(WorkerDict pid=364993)[0m 31
|
| 304 |
+
[36m(WorkerDict pid=364993)[0m ],
|
| 305 |
+
[36m(WorkerDict pid=364993)[0m "hidden_act": "silu",
|
| 306 |
+
[36m(WorkerDict pid=364993)[0m "hidden_size": 1280,
|
| 307 |
+
[36m(WorkerDict pid=364993)[0m "in_channels": 3,
|
| 308 |
+
[36m(WorkerDict pid=364993)[0m "in_chans": 3,
|
| 309 |
+
[36m(WorkerDict pid=364993)[0m "initializer_range": 0.02,
|
| 310 |
+
[36m(WorkerDict pid=364993)[0m "intermediate_size": 3420,
|
| 311 |
+
[36m(WorkerDict pid=364993)[0m "model_type": "qwen2_5_vl",
|
| 312 |
+
[36m(WorkerDict pid=364993)[0m "num_heads": 16,
|
| 313 |
+
[36m(WorkerDict pid=364993)[0m "out_hidden_size": 2048,
|
| 314 |
+
[36m(WorkerDict pid=364993)[0m "patch_size": 14,
|
| 315 |
+
[36m(WorkerDict pid=364993)[0m "spatial_merge_size": 2,
|
| 316 |
+
[36m(WorkerDict pid=364993)[0m "spatial_patch_size": 14,
|
| 317 |
+
[36m(WorkerDict pid=364993)[0m "temporal_patch_size": 2,
|
| 318 |
+
[36m(WorkerDict pid=364993)[0m "tokens_per_second": 2,
|
| 319 |
+
[36m(WorkerDict pid=364993)[0m "torch_dtype": "float32",
|
| 320 |
+
[36m(WorkerDict pid=364993)[0m "window_size": 112
|
| 321 |
+
[36m(WorkerDict pid=364993)[0m },
|
| 322 |
+
[36m(WorkerDict pid=364993)[0m "vision_end_token_id": 151653,
|
| 323 |
+
[36m(WorkerDict pid=364993)[0m "vision_start_token_id": 151652,
|
| 324 |
+
[36m(WorkerDict pid=364993)[0m "vision_token_id": 151654,
|
| 325 |
+
[36m(WorkerDict pid=364993)[0m "vocab_size": 151936
|
| 326 |
+
[36m(WorkerDict pid=364993)[0m }
|
| 327 |
+
[36m(WorkerDict pid=364993)[0m
|
| 328 |
+
Error executing job with overrides: ['algorithm.adv_estimator=grpo', 'data.train_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_full.parquet', 'data.val_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/test_full.parquet', 'data.train_batch_size=16', 'data.max_prompt_length=1024', 'data.max_response_length=1024', 'data.filter_overlong_prompts=True', 'data.truncation=error', 'data.image_key=images', 'actor_rollout_ref.model.path=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface', 'actor_rollout_ref.model.lora_rank=16', 'actor_rollout_ref.model.lora_alpha=16', 'actor_rollout_ref.model.lora_dropout=0.0', "actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'", 'actor_rollout_ref.actor.optim.lr=1e-4', 'actor_rollout_ref.model.use_remove_padding=False', 'actor_rollout_ref.actor.ppo_mini_batch_size=16', 'actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2', 'actor_rollout_ref.actor.use_kl_loss=True', 'actor_rollout_ref.actor.kl_loss_coef=0.001', 'actor_rollout_ref.actor.kl_loss_type=low_var_kl', 'actor_rollout_ref.model.enable_gradient_checkpointing=True', 'actor_rollout_ref.actor.fsdp_config.param_offload=False', 'actor_rollout_ref.actor.fsdp_config.optimizer_offload=False', 'actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.rollout.tensor_model_parallel_size=2', 'actor_rollout_ref.rollout.name=vllm', 'actor_rollout_ref.rollout.gpu_memory_utilization=0.4', 'actor_rollout_ref.rollout.enable_chunked_prefill=False', 'actor_rollout_ref.rollout.enforce_eager=False', 'actor_rollout_ref.rollout.free_cache_engine=False', 'actor_rollout_ref.rollout.n=5', 'actor_rollout_ref.rollout.val_kwargs.do_sample=False', 'actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.ref.fsdp_config.param_offload=True', 'algorithm.kl_ctrl.kl_coef=0.001', 'trainer.critic_warmup=0', "trainer.logger=['console']", 'trainer.project_name=anomseer', 'trainer.experiment_name=anomseer_rats_uni_full_20260612_184427_exp3_full_eval', 'trainer.default_local_dir=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3', 'trainer.n_gpus_per_node=2', 'trainer.nnodes=1', 'trainer.save_freq=500', 'trainer.test_freq=10', 'trainer.val_only=True', 'trainer.val_before_train=True', 'trainer.total_epochs=1', 'ts.use_sem_orth=True', 'ts.adv_mix=0.3', 'ts.similarity_method=ot', 'ts.ot_eps=0.08', 'ts.ot_n_iter=50', 'data.val_batch_size=64', 'trainer.test_freq=-1', 'trainer.val_before_train=True']
|
| 329 |
+
[36m(TaskRunner pid=363397)[0m Unhandled error (suppress with 'RAY_IGNORE_UNHANDLED_ERRORS=1'): [36mray::WorkerDict.ref_init_model()[39m (pid=365521, ip=172.20.115.25, actor_id=69baeda02fb711b165f2623201000000, repr=<verl.single_controller.ray.base.WorkerDict object at 0x7e5ac98e7d90>)
|
| 330 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 331 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 332 |
+
[36m(TaskRunner pid=363397)[0m File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 420, in func
|
| 333 |
+
[36m(TaskRunner pid=363397)[0m return getattr(self.worker_dict[key], name)(*args, **kwargs)
|
| 334 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 335 |
+
[36m(TaskRunner pid=363397)[0m File "/mnt/share01/sqk/AnomSeer/verl/single_controller/base/decorator.py", line 404, in inner
|
| 336 |
+
[36m(TaskRunner pid=363397)[0m return func(*args, **kwargs)
|
| 337 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^
|
| 338 |
+
[36m(TaskRunner pid=363397)[0m File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 466, in init_model
|
| 339 |
+
[36m(TaskRunner pid=363397)[0m self.ref_module_fsdp = self._build_model_optimizer(model_path=self.config.model.path,
|
| 340 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 341 |
+
[36m(TaskRunner pid=363397)[0m File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 217, in _build_model_optimizer
|
| 342 |
+
[36m(TaskRunner pid=363397)[0m actor_module = actor_module_class.from_pretrained(pretrained_model_name_or_path=local_path,
|
| 343 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 344 |
+
[36m(TaskRunner pid=363397)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/models/auto/modeling_auto.py", line 2165, in from_pretrained
|
| 345 |
+
[36m(TaskRunner pid=363397)[0m return super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
| 346 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 347 |
+
[36m(TaskRunner pid=363397)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/models/auto/auto_factory.py", line 600, in from_pretrained
|
| 348 |
+
[36m(TaskRunner pid=363397)[0m return model_class.from_pretrained(
|
| 349 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 350 |
+
[36m(TaskRunner pid=363397)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 315, in _wrapper
|
| 351 |
+
[36m(TaskRunner pid=363397)[0m return func(*args, **kwargs)
|
| 352 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^
|
| 353 |
+
[36m(TaskRunner pid=363397)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 4854, in from_pretrained
|
| 354 |
+
[36m(TaskRunner pid=363397)[0m checkpoint_files, sharded_metadata = _get_resolved_checkpoint_files(
|
| 355 |
+
[36m(TaskRunner pid=363397)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 356 |
+
[36m(TaskRunner pid=363397)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 1093, in _get_resolved_checkpoint_files
|
| 357 |
+
[36m(TaskRunner pid=363397)[0m raise OSError(
|
| 358 |
+
[36m(TaskRunner pid=363397)[0m OSError: Error no file named pytorch_model.bin, model.safetensors, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface.
|
| 359 |
+
Traceback (most recent call last):
|
| 360 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 68, in main
|
| 361 |
+
ray.get(runner.run.remote(config))
|
| 362 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/auto_init_hook.py", line 22, in auto_init_wrapper
|
| 363 |
+
return fn(*args, **kwargs)
|
| 364 |
+
^^^^^^^^^^^^^^^^^^^
|
| 365 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/client_mode_hook.py", line 107, in wrapper
|
| 366 |
+
return func(*args, **kwargs)
|
| 367 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 368 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 2980, in get
|
| 369 |
+
values, debugger_breakpoint = worker.get_objects(
|
| 370 |
+
^^^^^^^^^^^^^^^^^^^
|
| 371 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 1023, in get_objects
|
| 372 |
+
raise value.as_instanceof_cause()
|
| 373 |
+
ray.exceptions.RayTaskError(OSError): [36mray::TaskRunner.run()[39m (pid=363397, ip=172.20.115.25, actor_id=eb96e7c0e8f2d425a19fb2e901000000, repr=<main_ppo.TaskRunner object at 0x77b72dd69590>)
|
| 374 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 375 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 376 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 172, in run
|
| 377 |
+
trainer.init_workers()
|
| 378 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/ppo/ray_trainer.py", line 2024, in init_workers
|
| 379 |
+
self.ref_policy_wg.init_model()
|
| 380 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 43, in func
|
| 381 |
+
output = ray.get(output)
|
| 382 |
+
^^^^^^^^^^^^^^^
|
| 383 |
+
^^^^^^^^^^^^^^^^^^^
|
| 384 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 385 |
+
^^^^^^^^^^^^^^^^^^^
|
| 386 |
+
ray.exceptions.RayTaskError(OSError): [36mray::WorkerDict.ref_init_model()[39m (pid=364993, ip=172.20.115.25, actor_id=30e087db86b71d1893cef58101000000, repr=<verl.single_controller.ray.base.WorkerDict object at 0x73530b26be10>)
|
| 387 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 388 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 389 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 420, in func
|
| 390 |
+
return getattr(self.worker_dict[key], name)(*args, **kwargs)
|
| 391 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 392 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/base/decorator.py", line 404, in inner
|
| 393 |
+
return func(*args, **kwargs)
|
| 394 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 395 |
+
File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 466, in init_model
|
| 396 |
+
self.ref_module_fsdp = self._build_model_optimizer(model_path=self.config.model.path,
|
| 397 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 398 |
+
File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 217, in _build_model_optimizer
|
| 399 |
+
actor_module = actor_module_class.from_pretrained(pretrained_model_name_or_path=local_path,
|
| 400 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 401 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/models/auto/modeling_auto.py", line 2165, in from_pretrained
|
| 402 |
+
return super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
| 403 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 404 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/models/auto/auto_factory.py", line 600, in from_pretrained
|
| 405 |
+
return model_class.from_pretrained(
|
| 406 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 407 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 315, in _wrapper
|
| 408 |
+
return func(*args, **kwargs)
|
| 409 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 410 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 4854, in from_pretrained
|
| 411 |
+
checkpoint_files, sharded_metadata = _get_resolved_checkpoint_files(
|
| 412 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 413 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/transformers/modeling_utils.py", line 1093, in _get_resolved_checkpoint_files
|
| 414 |
+
raise OSError(
|
| 415 |
+
OSError: Error no file named pytorch_model.bin, model.safetensors, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface.
|
| 416 |
+
|
| 417 |
+
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
|
logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_122910.log
ADDED
|
@@ -0,0 +1,1196 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[log] saving full output to: /mnt/share01/sqk/AnomSeer/logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_122910.log
|
| 2 |
+
[config] tuning=LoRA (r=16, alpha=16, target=q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj) | lr=1e-4 | gpus=2 | tensor-parallel (TP=2)
|
| 3 |
+
[config] model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 4 |
+
[config] python=/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python | stage=eval
|
| 5 |
+
[stage] EVAL (model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface, gpus=2, tp=2, lora_rank=0)
|
| 6 |
+
2026-06-15 12:29:20,742 INFO worker.py:2012 -- Started a local Ray instance.
|
| 7 |
+
/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py:2051: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
|
| 8 |
+
warnings.warn(
|
| 9 |
+
[36m(TaskRunner pid=1868415)[0m {'actor_rollout_ref': {'actor': {'clip_ratio': 0.2,
|
| 10 |
+
[36m(TaskRunner pid=1868415)[0m 'entropy_coeff': 0.001,
|
| 11 |
+
[36m(TaskRunner pid=1868415)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 12 |
+
[36m(TaskRunner pid=1868415)[0m 'optimizer_offload': False,
|
| 13 |
+
[36m(TaskRunner pid=1868415)[0m 'param_offload': False,
|
| 14 |
+
[36m(TaskRunner pid=1868415)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 15 |
+
[36m(TaskRunner pid=1868415)[0m 'grad_clip': 1.0,
|
| 16 |
+
[36m(TaskRunner pid=1868415)[0m 'kl_loss_coef': 0.001,
|
| 17 |
+
[36m(TaskRunner pid=1868415)[0m 'kl_loss_type': 'low_var_kl',
|
| 18 |
+
[36m(TaskRunner pid=1868415)[0m 'optim': {'lr': 0.0001,
|
| 19 |
+
[36m(TaskRunner pid=1868415)[0m 'lr_warmup_steps': -1,
|
| 20 |
+
[36m(TaskRunner pid=1868415)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 21 |
+
[36m(TaskRunner pid=1868415)[0m 'min_lr_ratio': None,
|
| 22 |
+
[36m(TaskRunner pid=1868415)[0m 'total_training_steps': -1,
|
| 23 |
+
[36m(TaskRunner pid=1868415)[0m 'warmup_style': 'constant'},
|
| 24 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_epochs': 1,
|
| 25 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_max_token_len_per_gpu': 16384,
|
| 26 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_micro_batch_size': None,
|
| 27 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_micro_batch_size_per_gpu': 2,
|
| 28 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_mini_batch_size': 16,
|
| 29 |
+
[36m(TaskRunner pid=1868415)[0m 'shuffle': False,
|
| 30 |
+
[36m(TaskRunner pid=1868415)[0m 'strategy': 'fsdp',
|
| 31 |
+
[36m(TaskRunner pid=1868415)[0m 'ulysses_sequence_parallel_size': 1,
|
| 32 |
+
[36m(TaskRunner pid=1868415)[0m 'use_dynamic_bsz': False,
|
| 33 |
+
[36m(TaskRunner pid=1868415)[0m 'use_kl_loss': True,
|
| 34 |
+
[36m(TaskRunner pid=1868415)[0m 'use_torch_compile': True},
|
| 35 |
+
[36m(TaskRunner pid=1868415)[0m 'hybrid_engine': True,
|
| 36 |
+
[36m(TaskRunner pid=1868415)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 37 |
+
[36m(TaskRunner pid=1868415)[0m 'external_lib': None,
|
| 38 |
+
[36m(TaskRunner pid=1868415)[0m 'lora_alpha': 16,
|
| 39 |
+
[36m(TaskRunner pid=1868415)[0m 'lora_dropout': 0.0,
|
| 40 |
+
[36m(TaskRunner pid=1868415)[0m 'lora_rank': 0,
|
| 41 |
+
[36m(TaskRunner pid=1868415)[0m 'lora_target_modules': 'q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',
|
| 42 |
+
[36m(TaskRunner pid=1868415)[0m 'override_config': {},
|
| 43 |
+
[36m(TaskRunner pid=1868415)[0m 'path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 44 |
+
[36m(TaskRunner pid=1868415)[0m 'use_remove_padding': False},
|
| 45 |
+
[36m(TaskRunner pid=1868415)[0m 'ref': {'fsdp_config': {'param_offload': True,
|
| 46 |
+
[36m(TaskRunner pid=1868415)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 47 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 48 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_micro_batch_size': None,
|
| 49 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 50 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_use_dynamic_bsz': False,
|
| 51 |
+
[36m(TaskRunner pid=1868415)[0m 'ulysses_sequence_parallel_size': 1},
|
| 52 |
+
[36m(TaskRunner pid=1868415)[0m 'rollout': {'disable_log_stats': True,
|
| 53 |
+
[36m(TaskRunner pid=1868415)[0m 'do_sample': True,
|
| 54 |
+
[36m(TaskRunner pid=1868415)[0m 'dtype': 'bfloat16',
|
| 55 |
+
[36m(TaskRunner pid=1868415)[0m 'enable_chunked_prefill': False,
|
| 56 |
+
[36m(TaskRunner pid=1868415)[0m 'enforce_eager': False,
|
| 57 |
+
[36m(TaskRunner pid=1868415)[0m 'free_cache_engine': False,
|
| 58 |
+
[36m(TaskRunner pid=1868415)[0m 'gpu_memory_utilization': 0.4,
|
| 59 |
+
[36m(TaskRunner pid=1868415)[0m 'ignore_eos': False,
|
| 60 |
+
[36m(TaskRunner pid=1868415)[0m 'load_format': 'dummy_dtensor',
|
| 61 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 62 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_micro_batch_size': None,
|
| 63 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 64 |
+
[36m(TaskRunner pid=1868415)[0m 'log_prob_use_dynamic_bsz': False,
|
| 65 |
+
[36m(TaskRunner pid=1868415)[0m 'max_model_len': None,
|
| 66 |
+
[36m(TaskRunner pid=1868415)[0m 'max_num_batched_tokens': 8192,
|
| 67 |
+
[36m(TaskRunner pid=1868415)[0m 'max_num_seqs': 1024,
|
| 68 |
+
[36m(TaskRunner pid=1868415)[0m 'n': 5,
|
| 69 |
+
[36m(TaskRunner pid=1868415)[0m 'name': 'vllm',
|
| 70 |
+
[36m(TaskRunner pid=1868415)[0m 'prompt_length': 1024,
|
| 71 |
+
[36m(TaskRunner pid=1868415)[0m 'response_length': 1024,
|
| 72 |
+
[36m(TaskRunner pid=1868415)[0m 'temperature': 1.0,
|
| 73 |
+
[36m(TaskRunner pid=1868415)[0m 'tensor_model_parallel_size': 2,
|
| 74 |
+
[36m(TaskRunner pid=1868415)[0m 'top_k': -1,
|
| 75 |
+
[36m(TaskRunner pid=1868415)[0m 'top_p': 1,
|
| 76 |
+
[36m(TaskRunner pid=1868415)[0m 'use_fire_sampling': False,
|
| 77 |
+
[36m(TaskRunner pid=1868415)[0m 'val_kwargs': {'do_sample': False,
|
| 78 |
+
[36m(TaskRunner pid=1868415)[0m 'n': 1,
|
| 79 |
+
[36m(TaskRunner pid=1868415)[0m 'temperature': 0.4,
|
| 80 |
+
[36m(TaskRunner pid=1868415)[0m 'top_k': -1,
|
| 81 |
+
[36m(TaskRunner pid=1868415)[0m 'top_p': 1.0}}},
|
| 82 |
+
[36m(TaskRunner pid=1868415)[0m 'algorithm': {'adv_estimator': 'grpo',
|
| 83 |
+
[36m(TaskRunner pid=1868415)[0m 'gamma': 1.0,
|
| 84 |
+
[36m(TaskRunner pid=1868415)[0m 'kl_ctrl': {'kl_coef': 0.001, 'type': 'fixed'},
|
| 85 |
+
[36m(TaskRunner pid=1868415)[0m 'kl_penalty': 'kl',
|
| 86 |
+
[36m(TaskRunner pid=1868415)[0m 'lam': 1.0},
|
| 87 |
+
[36m(TaskRunner pid=1868415)[0m 'critic': {'cliprange_value': 0.5,
|
| 88 |
+
[36m(TaskRunner pid=1868415)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 89 |
+
[36m(TaskRunner pid=1868415)[0m 'forward_micro_batch_size': None,
|
| 90 |
+
[36m(TaskRunner pid=1868415)[0m 'forward_micro_batch_size_per_gpu': None,
|
| 91 |
+
[36m(TaskRunner pid=1868415)[0m 'grad_clip': 1.0,
|
| 92 |
+
[36m(TaskRunner pid=1868415)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 93 |
+
[36m(TaskRunner pid=1868415)[0m 'external_lib': None,
|
| 94 |
+
[36m(TaskRunner pid=1868415)[0m 'fsdp_config': {'fsdp_size': -1,[36m(TaskRunner pid=1868415)[0m DeprecationWarning: `ray.state.available_resources_per_node` is a private attribute and access will be removed in a future Ray version.
|
| 95 |
+
|
| 96 |
+
[36m(TaskRunner pid=1868415)[0m 'optimizer_offload': False,
|
| 97 |
+
[36m(TaskRunner pid=1868415)[0m 'param_offload': False,
|
| 98 |
+
[36m(TaskRunner pid=1868415)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 99 |
+
[36m(TaskRunner pid=1868415)[0m 'override_config': {},
|
| 100 |
+
[36m(TaskRunner pid=1868415)[0m 'path': '~/models/deepseek-llm-7b-chat',
|
| 101 |
+
[36m(TaskRunner pid=1868415)[0m 'tokenizer_path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 102 |
+
[36m(TaskRunner pid=1868415)[0m 'use_remove_padding': False},
|
| 103 |
+
[36m(TaskRunner pid=1868415)[0m 'optim': {'lr': 1e-05,
|
| 104 |
+
[36m(TaskRunner pid=1868415)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 105 |
+
[36m(TaskRunner pid=1868415)[0m 'min_lr_ratio': None,
|
| 106 |
+
[36m(TaskRunner pid=1868415)[0m 'total_training_steps': -1,
|
| 107 |
+
[36m(TaskRunner pid=1868415)[0m 'warmup_style': 'constant'},
|
| 108 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_epochs': 1,
|
| 109 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_max_token_len_per_gpu': 32768,
|
| 110 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_micro_batch_size': None,
|
| 111 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_micro_batch_size_per_gpu': None,
|
| 112 |
+
[36m(TaskRunner pid=1868415)[0m 'ppo_mini_batch_size': 16,
|
| 113 |
+
[36m(TaskRunner pid=1868415)[0m 'shuffle': False,
|
| 114 |
+
[36m(TaskRunner pid=1868415)[0m 'strategy': 'fsdp',
|
| 115 |
+
[36m(TaskRunner pid=1868415)[0m 'ulysses_sequence_parallel_size': 1,
|
| 116 |
+
[36m(TaskRunner pid=1868415)[0m 'use_dynamic_bsz': False},
|
| 117 |
+
[36m(TaskRunner pid=1868415)[0m 'custom_reward_function': {'name': 'compute_score', 'path': None},
|
| 118 |
+
[36m(TaskRunner pid=1868415)[0m 'data': {'filter_overlong_prompts': True,
|
| 119 |
+
[36m(TaskRunner pid=1868415)[0m 'image_key': 'images',
|
| 120 |
+
[36m(TaskRunner pid=1868415)[0m 'max_prompt_length': 1024,
|
| 121 |
+
[36m(TaskRunner pid=1868415)[0m 'max_response_length': 1024,
|
| 122 |
+
[36m(TaskRunner pid=1868415)[0m 'prompt_key': 'prompt',
|
| 123 |
+
[36m(TaskRunner pid=1868415)[0m 'return_raw_chat': False,
|
| 124 |
+
[36m(TaskRunner pid=1868415)[0m 'return_raw_input_ids': False,
|
| 125 |
+
[36m(TaskRunner pid=1868415)[0m 'shuffle': True,
|
| 126 |
+
[36m(TaskRunner pid=1868415)[0m 'tokenizer': None,
|
| 127 |
+
[36m(TaskRunner pid=1868415)[0m 'train_batch_size': 16,
|
| 128 |
+
[36m(TaskRunner pid=1868415)[0m 'train_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_full.parquet',
|
| 129 |
+
[36m(TaskRunner pid=1868415)[0m 'truncation': 'error',
|
| 130 |
+
[36m(TaskRunner pid=1868415)[0m 'val_batch_size': 64,
|
| 131 |
+
[36m(TaskRunner pid=1868415)[0m 'val_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/test_full.parquet'},
|
| 132 |
+
[36m(TaskRunner pid=1868415)[0m 'reward_model': {'enable': False,
|
| 133 |
+
[36m(TaskRunner pid=1868415)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 134 |
+
[36m(TaskRunner pid=1868415)[0m 'max_length': None,
|
| 135 |
+
[36m(TaskRunner pid=1868415)[0m 'micro_batch_size': None,
|
| 136 |
+
[36m(TaskRunner pid=1868415)[0m 'micro_batch_size_per_gpu': None,
|
| 137 |
+
[36m(TaskRunner pid=1868415)[0m 'model': {'external_lib': None,
|
| 138 |
+
[36m(TaskRunner pid=1868415)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 139 |
+
[36m(TaskRunner pid=1868415)[0m 'param_offload': False,
|
| 140 |
+
[36m(TaskRunner pid=1868415)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 141 |
+
[36m(TaskRunner pid=1868415)[0m 'input_tokenizer': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 142 |
+
[36m(TaskRunner pid=1868415)[0m 'path': '~/models/FsfairX-LLaMA3-RM-v0.1',
|
| 143 |
+
[36m(TaskRunner pid=1868415)[0m 'use_remove_padding': False},
|
| 144 |
+
[36m(TaskRunner pid=1868415)[0m 'reward_manager': 'naive',
|
| 145 |
+
[36m(TaskRunner pid=1868415)[0m 'strategy': 'fsdp',
|
| 146 |
+
[36m(TaskRunner pid=1868415)[0m 'ulysses_sequence_parallel_size': 1,
|
| 147 |
+
[36m(TaskRunner pid=1868415)[0m 'use_dynamic_bsz': False},
|
| 148 |
+
[36m(TaskRunner pid=1868415)[0m 'trainer': {'balance_batch': True,
|
| 149 |
+
[36m(TaskRunner pid=1868415)[0m 'critic_warmup': 0,
|
| 150 |
+
[36m(TaskRunner pid=1868415)[0m 'default_hdfs_dir': None,
|
| 151 |
+
[36m(TaskRunner pid=1868415)[0m 'default_local_dir': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3',
|
| 152 |
+
[36m(TaskRunner pid=1868415)[0m 'del_local_ckpt_after_load': False,
|
| 153 |
+
[36m(TaskRunner pid=1868415)[0m 'experiment_name': 'anomseer_rats_uni_full_20260612_184427_exp3_full_eval',
|
| 154 |
+
[36m(TaskRunner pid=1868415)[0m 'logger': ['console'],
|
| 155 |
+
[36m(TaskRunner pid=1868415)[0m 'n_gpus_per_node': 2,
|
| 156 |
+
[36m(TaskRunner pid=1868415)[0m 'nnodes': 1,
|
| 157 |
+
[36m(TaskRunner pid=1868415)[0m 'project_name': 'anomseer',
|
| 158 |
+
[36m(TaskRunner pid=1868415)[0m 'remove_previous_ckpt_in_save': False,
|
| 159 |
+
[36m(TaskRunner pid=1868415)[0m 'resume_from_path': False,
|
| 160 |
+
[36m(TaskRunner pid=1868415)[0m 'resume_mode': 'disable',
|
| 161 |
+
[36m(TaskRunner pid=1868415)[0m 'save_freq': 500,
|
| 162 |
+
[36m(TaskRunner pid=1868415)[0m 'test_freq': -1,
|
| 163 |
+
[36m(TaskRunner pid=1868415)[0m 'total_epochs': 1,
|
| 164 |
+
[36m(TaskRunner pid=1868415)[0m 'total_training_steps': None,
|
| 165 |
+
[36m(TaskRunner pid=1868415)[0m 'val_before_train': True,
|
| 166 |
+
[36m(TaskRunner pid=1868415)[0m 'val_generations_to_log_to_wandb': 0,
|
| 167 |
+
[36m(TaskRunner pid=1868415)[0m 'val_only': True},
|
| 168 |
+
[36m(TaskRunner pid=1868415)[0m 'ts': {'adv_mix': 0.3,
|
| 169 |
+
[36m(TaskRunner pid=1868415)[0m 'clip_pooling': 'mean',
|
| 170 |
+
[36m(TaskRunner pid=1868415)[0m 'clip_temperature': 1.0,
|
| 171 |
+
[36m(TaskRunner pid=1868415)[0m 'ot_eps': 0.08,
|
| 172 |
+
[36m(TaskRunner pid=1868415)[0m 'ot_n_iter': 50,
|
| 173 |
+
[36m(TaskRunner pid=1868415)[0m 'similarity_method': 'ot',
|
| 174 |
+
[36m(TaskRunner pid=1868415)[0m 'use_sem_orth': True}}
|
| 175 |
+
[36m(TaskRunner pid=1868415)[0m reward_manager_name: anomts
|
| 176 |
+
[36m(TaskRunner pid=1868415)[0m WARNING: val_batch_size is deprecated. Validation datasets are sent to inference engines as a whole batch, which will schedule the memory themselves.
|
| 177 |
+
[36m(TaskRunner pid=1868415)[0m [validate_config] All configuration checks passed successfully!
|
| 178 |
+
[36m(TaskRunner pid=1868415)[0m dataset len: 30266
|
| 179 |
+
[36m(TaskRunner pid=1868415)[0m filter dataset len: 30266
|
| 180 |
+
[36m(TaskRunner pid=1868415)[0m dataset len: 6034
|
| 181 |
+
[36m(TaskRunner pid=1868415)[0m filter dataset len: 6034
|
| 182 |
+
[36m(TaskRunner pid=1868415)[0m Size of train dataloader: 1891
|
| 183 |
+
[36m(TaskRunner pid=1868415)[0m Total training steps: 1891
|
| 184 |
+
[36m(WorkerDict pid=1875911)[0m Model config after override: Qwen2_5_VLConfig {
|
| 185 |
+
[36m(WorkerDict pid=1875911)[0m "architectures": [
|
| 186 |
+
[36m(WorkerDict pid=1875911)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 187 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 188 |
+
[36m(WorkerDict pid=1875911)[0m "attention_dropout": 0.0,
|
| 189 |
+
[36m(WorkerDict pid=1875911)[0m "eos_token_id": 151645,
|
| 190 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 191 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 2048,
|
| 192 |
+
[36m(WorkerDict pid=1875911)[0m "image_token_id": 151655,
|
| 193 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 194 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 11008,
|
| 195 |
+
[36m(WorkerDict pid=1875911)[0m "max_position_embeddings": 128000,
|
| 196 |
+
[36m(WorkerDict pid=1875911)[0m "max_window_layers": 70,
|
| 197 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl",
|
| 198 |
+
[36m(WorkerDict pid=1875911)[0m "num_attention_heads": 16,[36m(WorkerDict pid=1875911)[0m
|
| 199 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 200 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 201 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 202 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 203 |
+
[36m(WorkerDict pid=1875911)[0m [rank0]:[W615 12:30:45.599970048 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 0] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 204 |
+
|
| 205 |
+
[36m(WorkerDict pid=1875911)[0m "num_hidden_layers": 36,
|
| 206 |
+
[36m(WorkerDict pid=1875911)[0m "num_key_value_heads": 2,
|
| 207 |
+
[36m(WorkerDict pid=1875911)[0m "pad_token_id": 151643,
|
| 208 |
+
[36m(WorkerDict pid=1875911)[0m "rms_norm_eps": 1e-06,
|
| 209 |
+
[36m(WorkerDict pid=1875911)[0m "rope_scaling": {
|
| 210 |
+
[36m(WorkerDict pid=1875911)[0m "mrope_section": [
|
| 211 |
+
[36m(WorkerDict pid=1875911)[0m 16,
|
| 212 |
+
[36m(WorkerDict pid=1875911)[0m 24,
|
| 213 |
+
[36m(WorkerDict pid=1875911)[0m 24
|
| 214 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 215 |
+
[36m(WorkerDict pid=1875911)[0m "rope_type": "default",
|
| 216 |
+
[36m(WorkerDict pid=1875911)[0m "type": "default"
|
| 217 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 218 |
+
[36m(WorkerDict pid=1875911)[0m "rope_theta": 1000000.0,
|
| 219 |
+
[36m(WorkerDict pid=1875911)[0m "sliding_window": 32768,
|
| 220 |
+
[36m(WorkerDict pid=1875911)[0m "text_config": {
|
| 221 |
+
[36m(WorkerDict pid=1875911)[0m "architectures": [
|
| 222 |
+
[36m(WorkerDict pid=1875911)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 223 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 224 |
+
[36m(WorkerDict pid=1875911)[0m "attention_dropout": 0.0,
|
| 225 |
+
[36m(WorkerDict pid=1875911)[0m "bos_token_id": 151643,
|
| 226 |
+
[36m(WorkerDict pid=1875911)[0m "eos_token_id": 151645,
|
| 227 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 228 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 2048,
|
| 229 |
+
[36m(WorkerDict pid=1875911)[0m "image_token_id": null,
|
| 230 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 231 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 11008,
|
| 232 |
+
[36m(WorkerDict pid=1875911)[0m "layer_types": [
|
| 233 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 234 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 235 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 236 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 237 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 238 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 239 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 240 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 241 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 242 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 243 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 244 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 245 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 246 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 247 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 248 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 249 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 250 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 251 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 252 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 253 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 254 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 255 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 256 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 257 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 258 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 259 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 260 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 261 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 262 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 263 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 264 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 265 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 266 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 267 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 268 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention"
|
| 269 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 270 |
+
[36m(WorkerDict pid=1875911)[0m "max_position_embeddings": 128000,
|
| 271 |
+
[36m(WorkerDict pid=1875911)[0m "max_window_layers": 70,
|
| 272 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl_text",
|
| 273 |
+
[36m(WorkerDict pid=1875911)[0m "num_attention_heads": 16,
|
| 274 |
+
[36m(WorkerDict pid=1875911)[0m "num_hidden_layers": 36,
|
| 275 |
+
[36m(WorkerDict pid=1875911)[0m "num_key_value_heads": 2,
|
| 276 |
+
[36m(WorkerDict pid=1875911)[0m "rms_norm_eps": 1e-06,
|
| 277 |
+
[36m(WorkerDict pid=1875911)[0m "rope_scaling": {
|
| 278 |
+
[36m(WorkerDict pid=1875911)[0m "mrope_section": [
|
| 279 |
+
[36m(WorkerDict pid=1875911)[0m 16,
|
| 280 |
+
[36m(WorkerDict pid=1875911)[0m 24,
|
| 281 |
+
[36m(WorkerDict pid=1875911)[0m 24
|
| 282 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 283 |
+
[36m(WorkerDict pid=1875911)[0m "rope_type": "default",
|
| 284 |
+
[36m(WorkerDict pid=1875911)[0m "type": "default"
|
| 285 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 286 |
+
[36m(WorkerDict pid=1875911)[0m "rope_theta": 1000000.0,
|
| 287 |
+
[36m(WorkerDict pid=1875911)[0m "sliding_window": null,
|
| 288 |
+
[36m(WorkerDict pid=1875911)[0m "tie_word_embeddings": true,
|
| 289 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 290 |
+
[36m(WorkerDict pid=1875911)[0m "use_cache": true,
|
| 291 |
+
[36m(WorkerDict pid=1875911)[0m "use_sliding_window": false,
|
| 292 |
+
[36m(WorkerDict pid=1875911)[0m "video_token_id": null,
|
| 293 |
+
[36m(WorkerDict pid=1875911)[0m "vision_end_token_id": 151653,
|
| 294 |
+
[36m(WorkerDict pid=1875911)[0m "vision_start_token_id": 151652,
|
| 295 |
+
[36m(WorkerDict pid=1875911)[0m "vision_token_id": 151654,
|
| 296 |
+
[36m(WorkerDict pid=1875911)[0m "vocab_size": 151936
|
| 297 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 298 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 299 |
+
[36m(WorkerDict pid=1875911)[0m "transformers_version": "4.54.1",
|
| 300 |
+
[36m(WorkerDict pid=1875911)[0m "use_cache": true,
|
| 301 |
+
[36m(WorkerDict pid=1875911)[0m "use_sliding_window": false,
|
| 302 |
+
[36m(WorkerDict pid=1875911)[0m "video_token_id": 151656,
|
| 303 |
+
[36m(WorkerDict pid=1875911)[0m "vision_config": {
|
| 304 |
+
[36m(WorkerDict pid=1875911)[0m "depth": 32,
|
| 305 |
+
[36m(WorkerDict pid=1875911)[0m "fullatt_block_indexes": [
|
| 306 |
+
[36m(WorkerDict pid=1875911)[0m 7,
|
| 307 |
+
[36m(WorkerDict pid=1875911)[0m 15,
|
| 308 |
+
[36m(WorkerDict pid=1875911)[0m 23,
|
| 309 |
+
[36m(WorkerDict pid=1875911)[0m 31
|
| 310 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 311 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 312 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 1280,
|
| 313 |
+
[36m(WorkerDict pid=1875911)[0m "in_channels": 3,
|
| 314 |
+
[36m(WorkerDict pid=1875911)[0m "in_chans": 3,
|
| 315 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 316 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 3420,
|
| 317 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl",
|
| 318 |
+
[36m(WorkerDict pid=1875911)[0m "num_heads": 16,
|
| 319 |
+
[36m(WorkerDict pid=1875911)[0m "out_hidden_size": 2048,
|
| 320 |
+
[36m(WorkerDict pid=1875911)[0m "patch_size": 14,
|
| 321 |
+
[36m(WorkerDict pid=1875911)[0m "spatial_merge_size": 2,
|
| 322 |
+
[36m(WorkerDict pid=1875911)[0m "spatial_patch_size": 14,
|
| 323 |
+
[36m(WorkerDict pid=1875911)[0m "temporal_patch_size": 2,
|
| 324 |
+
[36m(WorkerDict pid=1875911)[0m "tokens_per_second": 2,
|
| 325 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 326 |
+
[36m(WorkerDict pid=1875911)[0m "window_size": 112
|
| 327 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 328 |
+
[36m(WorkerDict pid=1875911)[0m "vision_end_token_id": 151653,
|
| 329 |
+
[36m(WorkerDict pid=1875911)[0m "vision_start_token_id": 151652,
|
| 330 |
+
[36m(WorkerDict pid=1875911)[0m "vision_token_id": 151654,
|
| 331 |
+
[36m(WorkerDict pid=1875911)[0m "vocab_size": 151936
|
| 332 |
+
[36m(WorkerDict pid=1875911)[0m }
|
| 333 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 334 |
+
[36m(WorkerDict pid=1875911)[0m NCCL version 2.21.5+cuda12.4
|
| 335 |
+
[36m(WorkerDict pid=1875911)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 336 |
+
[36m(WorkerDict pid=1875911)[0m wrap_policy: functools.partial(<function _or_policy at 0x75feda0d8680>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x75feda0d8540>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 337 |
+
[36m(WorkerDict pid=1875911)[0m Actor use_remove_padding=False
|
| 338 |
+
[36m(WorkerDict pid=1877925)[0m wrap_policy: functools.partial(<function _or_policy at 0x71cf4e4c8680>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x71cf4e4c8540>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 339 |
+
[36m(WorkerDict pid=1875911)[0m Model config after override: Qwen2_5_VLConfig {
|
| 340 |
+
[36m(WorkerDict pid=1875911)[0m "architectures": [
|
| 341 |
+
[36m(WorkerDict pid=1875911)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 342 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 343 |
+
[36m(WorkerDict pid=1875911)[0m "attention_dropout": 0.0,
|
| 344 |
+
[36m(WorkerDict pid=1875911)[0m "eos_token_id": 151645,
|
| 345 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 346 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 2048,
|
| 347 |
+
[36m(WorkerDict pid=1875911)[0m "image_token_id": 151655,
|
| 348 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 349 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 11008,
|
| 350 |
+
[36m(WorkerDict pid=1875911)[0m "max_position_embeddings": 128000,
|
| 351 |
+
[36m(WorkerDict pid=1875911)[0m "max_window_layers": 70,
|
| 352 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl",
|
| 353 |
+
[36m(WorkerDict pid=1875911)[0m "num_attention_heads": 16,
|
| 354 |
+
[36m(WorkerDict pid=1875911)[0m "num_hidden_layers": 36,
|
| 355 |
+
[36m(WorkerDict pid=1875911)[0m "num_key_value_heads": 2,
|
| 356 |
+
[36m(WorkerDict pid=1875911)[0m "pad_token_id": 151643,
|
| 357 |
+
[36m(WorkerDict pid=1875911)[0m "rms_norm_eps": 1e-06,
|
| 358 |
+
[36m(WorkerDict pid=1875911)[0m "rope_scaling": {
|
| 359 |
+
[36m(WorkerDict pid=1875911)[0m "mrope_section": [
|
| 360 |
+
[36m(WorkerDict pid=1875911)[0m 16,
|
| 361 |
+
[36m(WorkerDict pid=1875911)[0m 24,
|
| 362 |
+
[36m(WorkerDict pid=1875911)[0m 24
|
| 363 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 364 |
+
[36m(WorkerDict pid=1875911)[0m "rope_type": "default",
|
| 365 |
+
[36m(WorkerDict pid=1875911)[0m "type": "default"
|
| 366 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 367 |
+
[36m(WorkerDict pid=1875911)[0m "rope_theta": 1000000.0,
|
| 368 |
+
[36m(WorkerDict pid=1875911)[0m "sliding_window": 32768,
|
| 369 |
+
[36m(WorkerDict pid=1875911)[0m "text_config": {
|
| 370 |
+
[36m(WorkerDict pid=1875911)[0m "architectures": [
|
| 371 |
+
[36m(WorkerDict pid=1875911)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 372 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 373 |
+
[36m(WorkerDict pid=1875911)[0m "attention_dropout": 0.0,
|
| 374 |
+
[36m(WorkerDict pid=1875911)[0m "bos_token_id": 151643,
|
| 375 |
+
[36m(WorkerDict pid=1875911)[0m "eos_token_id": 151645,
|
| 376 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 377 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 2048,
|
| 378 |
+
[36m(WorkerDict pid=1875911)[0m "image_token_id": null,
|
| 379 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 380 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 11008,
|
| 381 |
+
[36m(WorkerDict pid=1875911)[0m "layer_types": [
|
| 382 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 383 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 384 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 385 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 386 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 387 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 388 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 389 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 390 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 391 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 392 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 393 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 394 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 395 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 396 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 397 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 398 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 399 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 400 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 401 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 402 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 403 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 404 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 405 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 406 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 407 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 408 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 409 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 410 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 411 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 412 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 413 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 414 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 415 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 416 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention",
|
| 417 |
+
[36m(WorkerDict pid=1875911)[0m "full_attention"
|
| 418 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 419 |
+
[36m(WorkerDict pid=1875911)[0m "max_position_embeddings": 128000,
|
| 420 |
+
[36m(WorkerDict pid=1875911)[0m "max_window_layers": 70,
|
| 421 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl_text",
|
| 422 |
+
[36m(WorkerDict pid=1875911)[0m "num_attention_heads": 16,
|
| 423 |
+
[36m(WorkerDict pid=1875911)[0m "num_hidden_layers": 36,
|
| 424 |
+
[36m(WorkerDict pid=1875911)[0m "num_key_value_heads": 2,
|
| 425 |
+
[36m(WorkerDict pid=1875911)[0m "rms_norm_eps": 1e-06,
|
| 426 |
+
[36m(WorkerDict pid=1875911)[0m "rope_scaling": {
|
| 427 |
+
[36m(WorkerDict pid=1875911)[0m "mrope_section": [
|
| 428 |
+
[36m(WorkerDict pid=1875911)[0m 16,
|
| 429 |
+
[36m(WorkerDict pid=1875911)[0m 24,
|
| 430 |
+
[36m(WorkerDict pid=1875911)[0m 24
|
| 431 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 432 |
+
[36m(WorkerDict pid=1875911)[0m "rope_type": "default",
|
| 433 |
+
[36m(WorkerDict pid=1875911)[0m "type": "default"
|
| 434 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 435 |
+
[36m(WorkerDict pid=1875911)[0m "rope_theta": 1000000.0,
|
| 436 |
+
[36m(WorkerDict pid=1875911)[0m "sliding_window": null,
|
| 437 |
+
[36m(WorkerDict pid=1875911)[0m "tie_word_embeddings": true,
|
| 438 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 439 |
+
[36m(WorkerDict pid=1875911)[0m "use_cache": true,
|
| 440 |
+
[36m(WorkerDict pid=1875911)[0m "use_sliding_window": false,
|
| 441 |
+
[36m(WorkerDict pid=1875911)[0m "video_token_id": null,
|
| 442 |
+
[36m(WorkerDict pid=1875911)[0m "vision_end_token_id": 151653,
|
| 443 |
+
[36m(WorkerDict pid=1875911)[0m "vision_start_token_id": 151652,
|
| 444 |
+
[36m(WorkerDict pid=1875911)[0m "vision_token_id": 151654,
|
| 445 |
+
[36m(WorkerDict pid=1875911)[0m "vocab_size": 151936
|
| 446 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 447 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 448 |
+
[36m(WorkerDict pid=1875911)[0m "transformers_version": "4.54.1",
|
| 449 |
+
[36m(WorkerDict pid=1875911)[0m "use_cache": true,
|
| 450 |
+
[36m(WorkerDict pid=1875911)[0m "use_sliding_window": false,
|
| 451 |
+
[36m(WorkerDict pid=1875911)[0m "video_token_id": 151656,
|
| 452 |
+
[36m(WorkerDict pid=1875911)[0m "vision_config": {
|
| 453 |
+
[36m(WorkerDict pid=1875911)[0m "depth": 32,
|
| 454 |
+
[36m(WorkerDict pid=1875911)[0m "fullatt_block_indexes": [
|
| 455 |
+
[36m(WorkerDict pid=1875911)[0m 7,
|
| 456 |
+
[36m(WorkerDict pid=1875911)[0m 15,
|
| 457 |
+
[36m(WorkerDict pid=1875911)[0m 23,
|
| 458 |
+
[36m(WorkerDict pid=1875911)[0m 31
|
| 459 |
+
[36m(WorkerDict pid=1875911)[0m ],
|
| 460 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_act": "silu",
|
| 461 |
+
[36m(WorkerDict pid=1875911)[0m "hidden_size": 1280,
|
| 462 |
+
[36m(WorkerDict pid=1875911)[0m "in_channels": 3,
|
| 463 |
+
[36m(WorkerDict pid=1875911)[0m "in_chans": 3,
|
| 464 |
+
[36m(WorkerDict pid=1875911)[0m "initializer_range": 0.02,
|
| 465 |
+
[36m(WorkerDict pid=1875911)[0m "intermediate_size": 3420,
|
| 466 |
+
[36m(WorkerDict pid=1875911)[0m "model_type": "qwen2_5_vl",
|
| 467 |
+
[36m(WorkerDict pid=1875911)[0m "num_heads": 16,
|
| 468 |
+
[36m(WorkerDict pid=1875911)[0m "out_hidden_size": 2048,
|
| 469 |
+
[36m(WorkerDict pid=1875911)[0m "patch_size": 14,
|
| 470 |
+
[36m(WorkerDict pid=1875911)[0m "spatial_merge_size": 2,
|
| 471 |
+
[36m(WorkerDict pid=1875911)[0m "spatial_patch_size": 14,
|
| 472 |
+
[36m(WorkerDict pid=1875911)[0m "temporal_patch_size": 2,
|
| 473 |
+
[36m(WorkerDict pid=1875911)[0m "tokens_per_second": 2,
|
| 474 |
+
[36m(WorkerDict pid=1875911)[0m "torch_dtype": "float32",
|
| 475 |
+
[36m(WorkerDict pid=1875911)[0m "window_size": 112
|
| 476 |
+
[36m(WorkerDict pid=1875911)[0m },
|
| 477 |
+
[36m(WorkerDict pid=1875911)[0m "vision_end_token_id": 151653,
|
| 478 |
+
[36m(WorkerDict pid=1875911)[0m "vision_start_token_id": 151652,
|
| 479 |
+
[36m(WorkerDict pid=1875911)[0m "vision_token_id": 151654,[36m(WorkerDict pid=1875911)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLForConditionalGeneration is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 480 |
+
[36m(WorkerDict pid=1875911)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 481 |
+
[36m(WorkerDict pid=1875911)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VisionTransformerPretrainedModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 482 |
+
[36m(WorkerDict pid=1875911)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 483 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 484 |
+
[36m(WorkerDict pid=1877925)[0m [rank1]:[W615 12:30:45.595731994 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 1] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 485 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 486 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 487 |
+
[36m(WorkerDict pid=1877925)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`[32m [repeated 4x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/user-guides/configure-logging.html#log-deduplication for more options.)[0m
|
| 488 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 489 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 490 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 491 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 492 |
+
[36m(WorkerDict pid=1877925)[0m
|
| 493 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 494 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 495 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 496 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 497 |
+
[36m(WorkerDict pid=1875911)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 498 |
+
[36m(WorkerDict pid=1875911)[0m warnings.warn(
|
| 499 |
+
|
| 500 |
+
[36m(WorkerDict pid=1875911)[0m "vocab_size": 151936
|
| 501 |
+
[36m(WorkerDict pid=1875911)[0m }
|
| 502 |
+
[36m(WorkerDict pid=1875911)[0m
|
| 503 |
+
[36m(WorkerDict pid=1875911)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 504 |
+
[36m(WorkerDict pid=1875911)[0m wrap_policy: functools.partial(<function _or_policy at 0x75feda0d8680>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x75feda0d8540>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 505 |
+
[36m(WorkerDict pid=1877925)[0m Actor use_remove_padding=False
|
| 506 |
+
[36m(WorkerDict pid=1875911)[0m Total steps: 1891, num_warmup_steps: 0
|
| 507 |
+
[36m(WorkerDict pid=1875911)[0m Actor use_remove_padding=False
|
| 508 |
+
[36m(WorkerDict pid=1875911)[0m Before building vllm rollout, memory allocated (GB): 7.027174949645996, memory reserved (GB): 12.189453125
|
| 509 |
+
[36m(WorkerDict pid=1875911)[0m WARNING 06-15 12:33:19 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x75fad74a9210>
|
| 510 |
+
[36m(WorkerDict pid=1877925)[0m wrap_policy: functools.partial(<function _or_policy at 0x71cf4e4c8680>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x71cf4e4c8540>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 511 |
+
[36m(WorkerDict pid=1877925)[0m Total steps: 1891, num_warmup_steps: 0
|
| 512 |
+
[36m(WorkerDict pid=1877925)[0m Actor use_remove_padding=False
|
| 513 |
+
[36m(WorkerDict pid=1875911)[0m WARNING 06-15 12:33:24 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 514 |
+
[36m(WorkerDict pid=1877925)[0m WARNING 06-15 12:33:19 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x71cbd0b48850>
|
| 515 |
+
[36m(WorkerDict pid=1875911)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 516 |
+
[36m(WorkerDict pid=1875911)[0m After building vllm rollout, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 517 |
+
[36m(WorkerDict pid=1875911)[0m After building sharding manager, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 518 |
+
[36m(WorkerDict pid=1877925)[0m WARNING 06-15 12:33:24 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 519 |
+
[36m(TaskRunner pid=1868415)[0m Using LocalLogger is deprecated. The constructor API will change
|
| 520 |
+
[36m(TaskRunner pid=1868415)[0m validation generation end
|
| 521 |
+
[36m(WorkerDict pid=1877925)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 522 |
+
[36m(TaskRunner pid=1868415)[0m [prompt] <|im_start|>system
|
| 523 |
+
[36m(TaskRunner pid=1868415)[0m You are a helpful assistant.<|im_end|>
|
| 524 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>user
|
| 525 |
+
[36m(TaskRunner pid=1868415)[0m <|vision_start|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|vision_end|>
|
| 526 |
+
[36m(TaskRunner pid=1868415)[0m You are an expert in univariate time-series anomaly detection. The figure shows a single-channel time series of length 128 from the "Healthcare-ECG" domain.
|
| 527 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 528 |
+
[36m(TaskRunner pid=1868415)[0m Decide whether the series is normal or contains an anomaly. If it is anomalous, choose the single most appropriate anomaly type from the following 15 categories (format `id: name — description`):
|
| 529 |
+
[36m(TaskRunner pid=1868415)[0m 0: Normal Sequence — There are no abnormal situations in this time series.
|
| 530 |
+
[36m(TaskRunner pid=1868415)[0m 1: Point Anomaly — A single data point significantly deviates from the local or global pattern of the sequence.
|
| 531 |
+
[36m(TaskRunner pid=1868415)[0m 2: Periodic Change Anomaly — The original periodic pattern is disrupted, e.g. the period is broken or the amplitude becomes anomalous.
|
| 532 |
+
[36m(TaskRunner pid=1868415)[0m 3: Trend Change Anomaly — A sudden change in the long-term trend of the time series.
|
| 533 |
+
[36m(TaskRunner pid=1868415)[0m 4: Change Point Anomaly — Statistical properties (e.g. mean, variance) change abruptly at certain points.
|
| 534 |
+
[36m(TaskRunner pid=1868415)[0m 5: Distributional Change Anomaly — The statistical distribution of the time series changes significantly.
|
| 535 |
+
[36m(TaskRunner pid=1868415)[0m 6: Amplitude Anomaly — The amplitude of data points exceeds the normal upper and lower bounds.
|
| 536 |
+
[36m(TaskRunner pid=1868415)[0m 7: Pattern Change Anomaly — The pattern of the time series suddenly changes from one form to another.
|
| 537 |
+
[36m(TaskRunner pid=1868415)[0m 8: Sparse Anomaly — Isolated anomalous patterns occasionally appear in a long time series.
|
| 538 |
+
[36m(TaskRunner pid=1868415)[0m 9: Repeated Value Anomaly — Continuous or intermittent repeated values disrupt the normal fluctuation pattern.
|
| 539 |
+
[36m(TaskRunner pid=1868415)[0m 10: Sudden Flatline Anomaly — The time series suddenly becomes a flat line with no normal fluctuations.
|
| 540 |
+
[36m(TaskRunner pid=1868415)[0m 11: Drift Anomaly — The data gradually drifts away from the normal level.
|
| 541 |
+
[36m(TaskRunner pid=1868415)[0m 12: Sudden Spike Anomaly — The data suddenly spikes or drops within a short time and then returns to normal.
|
| 542 |
+
[36m(TaskRunner pid=1868415)[0m 13: Continuous Segment Anomaly — A continuous segment of data points deviates from the normal pattern.
|
| 543 |
+
[36m(TaskRunner pid=1868415)[0m 14: Nonlinear Pattern Anomaly — Nonlinear changes appear in the sequence, breaking the original linear rule.
|
| 544 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 545 |
+
[36m(TaskRunner pid=1868415)[0m Reason step by step inside <think>...</think> based on the visual shape of the series, then output exactly one line with your final answer:
|
| 546 |
+
[36m(TaskRunner pid=1868415)[0m <class>one exact category name from the list above</class>
|
| 547 |
+
[36m(TaskRunner pid=1868415)[0m If the series is normal, use <class>Normal Sequence</class>.<|im_end|>
|
| 548 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>assistant
|
| 549 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 550 |
+
[36m(TaskRunner pid=1868415)[0m [response] <think></think>
|
| 551 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 552 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 553 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 554 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 555 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 556 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 557 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 558 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 559 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 560 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 561 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 562 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 563 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 564 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 565 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 566 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 567 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 568 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 569 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 570 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 571 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 572 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 573 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 574 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 575 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 576 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 577 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 578 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 579 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 580 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 581 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 582 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 583 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 584 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 585 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 586 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 587 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 588 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 589 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 590 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 591 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 592 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 593 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 594 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 595 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 596 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 597 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 598 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 599 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 600 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 601 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 602 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 603 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 604 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 605 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 606 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 607 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 608 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 609 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 610 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 611 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 612 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 613 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 614 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 615 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 616 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 617 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 618 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 619 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 620 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 621 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 622 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 623 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 624 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 625 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 626 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 627 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 628 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 629 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 630 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 631 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 632 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 633 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 634 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 635 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 636 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 637 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 638 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 639 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 640 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 641 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 642 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 643 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 644 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 645 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 646 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 647 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 648 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 649 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 650 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 651 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 652 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 653 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 654 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 655 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 656 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 657 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 658 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 659 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 660 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 661 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 662 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 663 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 664 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 665 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 666 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 667 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 668 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 669 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 670 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 671 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 672 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 673 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 674 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 675 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 676 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 677 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 678 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 679 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 680 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 681 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 682 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 683 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 684 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 685 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 686 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 687 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 688 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 689 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 690 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 691 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 692 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 693 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 694 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 695 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 696 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 697 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 698 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 699 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 700 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 701 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 702 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 703 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 704 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 705 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 706 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 707 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 708 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 709 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 710 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 711 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 712 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 713 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 714 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 715 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 716 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 717 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 718 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 719 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 720 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 721 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 722 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 723 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 724 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 725 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 726 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 727 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 728 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 729 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 730 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 731 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 732 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 733 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 734 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 735 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 736 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 737 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 738 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 739 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 740 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 741 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 742 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 743 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 744 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 745 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 746 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 747 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 748 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 749 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 750 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 751 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 752 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 753 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 754 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 755 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 756 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 757 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 758 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 759 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 760 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 761 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 762 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 763 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 764 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 765 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 766 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 767 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 768 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 769 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 770 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 771 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 772 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 773 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 774 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 775 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 776 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 777 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 778 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 779 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 780 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 781 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 782 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 783 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 784 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 785 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 786 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 787 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 788 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 789 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 790 |
+
[36m(TaskRunner pid=1868415)[0m Detected new reward function format (3 return values).
|
| 791 |
+
[36m(TaskRunner pid=1868415)[0m validation generation end
|
| 792 |
+
[36m(TaskRunner pid=1868415)[0m [prompt] <|im_start|>system
|
| 793 |
+
[36m(TaskRunner pid=1868415)[0m You are a helpful assistant.<|im_end|>
|
| 794 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>user
|
| 795 |
+
[36m(TaskRunner pid=1868415)[0m <|vision_start|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|vision_end|>
|
| 796 |
+
[36m(TaskRunner pid=1868415)[0m You are an expert in univariate time-series anomaly detection. The figure shows a single-channel time series of length 32 from the "Server-YAHOO" domain.
|
| 797 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 798 |
+
[36m(TaskRunner pid=1868415)[0m Decide whether the series is normal or contains an anomaly. If it is anomalous, choose the single most appropriate anomaly type from the following 15 categories (format `id: name — description`):
|
| 799 |
+
[36m(TaskRunner pid=1868415)[0m 0: Normal Sequence — There are no abnormal situations in this time series.
|
| 800 |
+
[36m(TaskRunner pid=1868415)[0m 1: Point Anomaly — A single data point significantly deviates from the local or global pattern of the sequence.
|
| 801 |
+
[36m(TaskRunner pid=1868415)[0m 2: Periodic Change Anomaly — The original periodic pattern is disrupted, e.g. the period is broken or the amplitude becomes anomalous.
|
| 802 |
+
[36m(TaskRunner pid=1868415)[0m 3: Trend Change Anomaly — A sudden change in the long-term trend of the time series.
|
| 803 |
+
[36m(TaskRunner pid=1868415)[0m 4: Change Point Anomaly — Statistical properties (e.g. mean, variance) change abruptly at certain points.
|
| 804 |
+
[36m(TaskRunner pid=1868415)[0m 5: Distributional Change Anomaly — The statistical distribution of the time series changes significantly.
|
| 805 |
+
[36m(TaskRunner pid=1868415)[0m 6: Amplitude Anomaly — The amplitude of data points exceeds the normal upper and lower bounds.
|
| 806 |
+
[36m(TaskRunner pid=1868415)[0m 7: Pattern Change Anomaly — The pattern of the time series suddenly changes from one form to another.
|
| 807 |
+
[36m(TaskRunner pid=1868415)[0m 8: Sparse Anomaly — Isolated anomalous patterns occasionally appear in a long time series.
|
| 808 |
+
[36m(TaskRunner pid=1868415)[0m 9: Repeated Value Anomaly — Continuous or intermittent repeated values disrupt the normal fluctuation pattern.
|
| 809 |
+
[36m(TaskRunner pid=1868415)[0m 10: Sudden Flatline Anomaly — The time series suddenly becomes a flat line with no normal fluctuations.
|
| 810 |
+
[36m(TaskRunner pid=1868415)[0m 11: Drift Anomaly — The data gradually drifts away from the normal level.
|
| 811 |
+
[36m(TaskRunner pid=1868415)[0m 12: Sudden Spike Anomaly — The data suddenly spikes or drops within a short time and then returns to normal.
|
| 812 |
+
[36m(TaskRunner pid=1868415)[0m 13: Continuous Segment Anomaly — A continuous segment of data points deviates from the normal pattern.
|
| 813 |
+
[36m(TaskRunner pid=1868415)[0m 14: Nonlinear Pattern Anomaly — Nonlinear changes appear in the sequence, breaking the original linear rule.
|
| 814 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 815 |
+
[36m(TaskRunner pid=1868415)[0m Reason step by step inside <think>...</think> based on the visual shape of the series, then output exactly one line with your final answer:
|
| 816 |
+
[36m(TaskRunner pid=1868415)[0m <class>one exact category name from the list above</class>
|
| 817 |
+
[36m(TaskRunner pid=1868415)[0m If the series is normal, use <class>Normal Sequence</class>.<|im_end|>
|
| 818 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>assistant
|
| 819 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 820 |
+
[36m(TaskRunner pid=1868415)[0m [response] <think></think>
|
| 821 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 822 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 823 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 824 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 825 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 826 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 827 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 828 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 829 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 830 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 831 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 832 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 833 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 834 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 835 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 836 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 837 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 838 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 839 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 840 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 841 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 842 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 843 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 844 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 845 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 846 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 847 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 848 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 849 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 850 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 851 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 852 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 853 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 854 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 855 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 856 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 857 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 858 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 859 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 860 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 861 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 862 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 863 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 864 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 865 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 866 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 867 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 868 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 869 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 870 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 871 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 872 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 873 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 874 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 875 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 876 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 877 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 878 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 879 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 880 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 881 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 882 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 883 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 884 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 885 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 886 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 887 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 888 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 889 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 890 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 891 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 892 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 893 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 894 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 895 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 896 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 897 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 898 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 899 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 900 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 901 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 902 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 903 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 904 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 905 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 906 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 907 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 908 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 909 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 910 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 911 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 912 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 913 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 914 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 915 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 916 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 917 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 918 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 919 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 920 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 921 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 922 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 923 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 924 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 925 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 926 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 927 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 928 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 929 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 930 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 931 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 932 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 933 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 934 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 935 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 936 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 937 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 938 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 939 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 940 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 941 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 942 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 943 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 944 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 945 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 946 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 947 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 948 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 949 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 950 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 951 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 952 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 953 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 954 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 955 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 956 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 957 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 958 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 959 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 960 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 961 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 962 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 963 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 964 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 965 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 966 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 967 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 968 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 969 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 970 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 971 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 972 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 973 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 974 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 975 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 976 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 977 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 978 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 979 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 980 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 981 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 982 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 983 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 984 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 985 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 986 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 987 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 988 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 989 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 990 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 991 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 992 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 993 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 994 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 995 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 996 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 997 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 998 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 999 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1000 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1001 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1002 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1003 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1004 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1005 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1006 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1007 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1008 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1009 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1010 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1011 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1012 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1013 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1014 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1015 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1016 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1017 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1018 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1019 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1020 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1021 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1022 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1023 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1024 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1025 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1026 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1027 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1028 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1029 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1030 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1031 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1032 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1033 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1034 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1035 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1036 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1037 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1038 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1039 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1040 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1041 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1042 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1043 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1044 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1045 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1046 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1047 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1048 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1049 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1050 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1051 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1052 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1053 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1054 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1055 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1056 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1057 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1058 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1059 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1060 |
+
[36m(TaskRunner pid=1868415)[0m validation generation end
|
| 1061 |
+
[36m(TaskRunner pid=1868415)[0m [prompt] <|im_start|>system
|
| 1062 |
+
[36m(TaskRunner pid=1868415)[0m You are a helpful assistant.<|im_end|>
|
| 1063 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>user
|
| 1064 |
+
[36m(TaskRunner pid=1868415)[0m <|vision_start|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|vision_end|>
|
| 1065 |
+
[36m(TaskRunner pid=1868415)[0m You are an expert in univariate time-series anomaly detection. The figure shows a single-channel time series of length 128 from the "Healthcare-ECG" domain.
|
| 1066 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1067 |
+
[36m(TaskRunner pid=1868415)[0m Decide whether the series is normal or contains an anomaly. If it is anomalous, choose the single most appropriate anomaly type from the following 15 categories (format `id: name — description`):
|
| 1068 |
+
[36m(TaskRunner pid=1868415)[0m 0: Normal Sequence — There are no abnormal situations in this time series.
|
| 1069 |
+
[36m(TaskRunner pid=1868415)[0m 1: Point Anomaly — A single data point significantly deviates from the local or global pattern of the sequence.
|
| 1070 |
+
[36m(TaskRunner pid=1868415)[0m 2: Periodic Change Anomaly — The original periodic pattern is disrupted, e.g. the period is broken or the amplitude becomes anomalous.
|
| 1071 |
+
[36m(TaskRunner pid=1868415)[0m 3: Trend Change Anomaly — A sudden change in the long-term trend of the time series.
|
| 1072 |
+
[36m(TaskRunner pid=1868415)[0m 4: Change Point Anomaly — Statistical properties (e.g. mean, variance) change abruptly at certain points.
|
| 1073 |
+
[36m(TaskRunner pid=1868415)[0m 5: Distributional Change Anomaly — The statistical distribution of the time series changes significantly.
|
| 1074 |
+
[36m(TaskRunner pid=1868415)[0m 6: Amplitude Anomaly — The amplitude of data points exceeds the normal upper and lower bounds.
|
| 1075 |
+
[36m(TaskRunner pid=1868415)[0m 7: Pattern Change Anomaly — The pattern of the time series suddenly changes from one form to another.
|
| 1076 |
+
[36m(TaskRunner pid=1868415)[0m 8: Sparse Anomaly — Isolated anomalous patterns occasionally appear in a long time series.
|
| 1077 |
+
[36m(TaskRunner pid=1868415)[0m 9: Repeated Value Anomaly — Continuous or intermittent repeated values disrupt the normal fluctuation pattern.
|
| 1078 |
+
[36m(TaskRunner pid=1868415)[0m 10: Sudden Flatline Anomaly — The time series suddenly becomes a flat line with no normal fluctuations.
|
| 1079 |
+
[36m(TaskRunner pid=1868415)[0m 11: Drift Anomaly — The data gradually drifts away from the normal level.
|
| 1080 |
+
[36m(TaskRunner pid=1868415)[0m 12: Sudden Spike Anomaly — The data suddenly spikes or drops within a short time and then returns to normal.
|
| 1081 |
+
[36m(TaskRunner pid=1868415)[0m 13: Continuous Segment Anomaly — A continuous segment of data points deviates from the normal pattern.
|
| 1082 |
+
[36m(TaskRunner pid=1868415)[0m 14: Nonlinear Pattern Anomaly — Nonlinear changes appear in the sequence, breaking the original linear rule.
|
| 1083 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1084 |
+
[36m(TaskRunner pid=1868415)[0m Reason step by step inside <think>...</think> based on the visual shape of the series, then output exactly one line with your final answer:
|
| 1085 |
+
[36m(TaskRunner pid=1868415)[0m <class>one exact category name from the list above</class>
|
| 1086 |
+
[36m(TaskRunner pid=1868415)[0m If the series is normal, use <class>Normal Sequence</class>.<|im_end|>
|
| 1087 |
+
[36m(TaskRunner pid=1868415)[0m <|im_start|>assistant
|
| 1088 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1089 |
+
[36m(TaskRunner pid=1868415)[0m [response] <think></think>
|
| 1090 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1091 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1092 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1093 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1094 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1095 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1096 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1097 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1098 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1099 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1100 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1101 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1102 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1103 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1104 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1105 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1106 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1107 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1108 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1109 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1110 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1111 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1112 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1113 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1114 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1115 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1116 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1117 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1118 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1119 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1120 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1121 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1122 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1123 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1124 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1125 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1126 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1127 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1128 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1129 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1130 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1131 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1132 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1133 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1134 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1135 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1136 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1137 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1138 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1139 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1140 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1141 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1142 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1143 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1144 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1145 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1146 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1147 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1148 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1149 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1150 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1151 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1152 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1153 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1154 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1155 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1156 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1157 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1158 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1159 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1160 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1161 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1162 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1163 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1164 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1165 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1166 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1167 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1168 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1169 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1170 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1171 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1172 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1173 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1174 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1175 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1176 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1177 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1178 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1179 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1180 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1181 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1182 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1183 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1184 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1185 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1186 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1187 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1188 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1189 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1190 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1191 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1192 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
| 1193 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1194 |
+
[36m(TaskRunner pid=1868415)[0m <class>Sudden Spike Anomaly</class>
|
| 1195 |
+
[36m(TaskRunner pid=1868415)[0m
|
| 1196 |
+
[36m(TaskRunner pid=1868415)[0m <think></think>
|
logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_20260615_154822.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_20260615.log
ADDED
|
@@ -0,0 +1,635 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[log] saving full output to: /mnt/share01/sqk/AnomSeer/logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_20260615.log
|
| 2 |
+
[config] tuning=LoRA (r=16, alpha=16, target=q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj) | lr=1e-4 | gpus=2 | tensor-parallel (TP=2)
|
| 3 |
+
[config] model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 4 |
+
[config] python=/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python | stage=eval
|
| 5 |
+
[stage] EVAL (model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface, gpus=2, tp=2, lora_rank=0)
|
| 6 |
+
2026-06-15 11:18:02,410 INFO worker.py:2012 -- Started a local Ray instance.
|
| 7 |
+
/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py:2051: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
|
| 8 |
+
warnings.warn(
|
| 9 |
+
[36m(TaskRunner pid=1165231)[0m {'actor_rollout_ref': {'actor': {'clip_ratio': 0.2,
|
| 10 |
+
[36m(TaskRunner pid=1165231)[0m 'entropy_coeff': 0.001,
|
| 11 |
+
[36m(TaskRunner pid=1165231)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 12 |
+
[36m(TaskRunner pid=1165231)[0m 'optimizer_offload': False,
|
| 13 |
+
[36m(TaskRunner pid=1165231)[0m 'param_offload': False,
|
| 14 |
+
[36m(TaskRunner pid=1165231)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 15 |
+
[36m(TaskRunner pid=1165231)[0m 'grad_clip': 1.0,
|
| 16 |
+
[36m(TaskRunner pid=1165231)[0m 'kl_loss_coef': 0.001,
|
| 17 |
+
[36m(TaskRunner pid=1165231)[0m 'kl_loss_type': 'low_var_kl',
|
| 18 |
+
[36m(TaskRunner pid=1165231)[0m 'optim': {'lr': 0.0001,
|
| 19 |
+
[36m(TaskRunner pid=1165231)[0m 'lr_warmup_steps': -1,
|
| 20 |
+
[36m(TaskRunner pid=1165231)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 21 |
+
[36m(TaskRunner pid=1165231)[0m 'min_lr_ratio': None,
|
| 22 |
+
[36m(TaskRunner pid=1165231)[0m 'total_training_steps': -1,
|
| 23 |
+
[36m(TaskRunner pid=1165231)[0m 'warmup_style': 'constant'},
|
| 24 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_epochs': 1,
|
| 25 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_max_token_len_per_gpu': 16384,
|
| 26 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_micro_batch_size': None,
|
| 27 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_micro_batch_size_per_gpu': 2,
|
| 28 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_mini_batch_size': 16,
|
| 29 |
+
[36m(TaskRunner pid=1165231)[0m 'shuffle': False,
|
| 30 |
+
[36m(TaskRunner pid=1165231)[0m 'strategy': 'fsdp',
|
| 31 |
+
[36m(TaskRunner pid=1165231)[0m 'ulysses_sequence_parallel_size': 1,
|
| 32 |
+
[36m(TaskRunner pid=1165231)[0m 'use_dynamic_bsz': False,
|
| 33 |
+
[36m(TaskRunner pid=1165231)[0m 'use_kl_loss': True,
|
| 34 |
+
[36m(TaskRunner pid=1165231)[0m 'use_torch_compile': True},
|
| 35 |
+
[36m(TaskRunner pid=1165231)[0m 'hybrid_engine': True,
|
| 36 |
+
[36m(TaskRunner pid=1165231)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 37 |
+
[36m(TaskRunner pid=1165231)[0m 'external_lib': None,
|
| 38 |
+
[36m(TaskRunner pid=1165231)[0m 'lora_alpha': 16,
|
| 39 |
+
[36m(TaskRunner pid=1165231)[0m 'lora_dropout': 0.0,
|
| 40 |
+
[36m(TaskRunner pid=1165231)[0m 'lora_rank': 0,
|
| 41 |
+
[36m(TaskRunner pid=1165231)[0m 'lora_target_modules': 'q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',
|
| 42 |
+
[36m(TaskRunner pid=1165231)[0m 'override_config': {},
|
| 43 |
+
[36m(TaskRunner pid=1165231)[0m 'path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 44 |
+
[36m(TaskRunner pid=1165231)[0m 'use_remove_padding': False},
|
| 45 |
+
[36m(TaskRunner pid=1165231)[0m 'ref': {'fsdp_config': {'param_offload': True,
|
| 46 |
+
[36m(TaskRunner pid=1165231)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 47 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 48 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_micro_batch_size': None,
|
| 49 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 50 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_use_dynamic_bsz': False,
|
| 51 |
+
[36m(TaskRunner pid=1165231)[0m 'ulysses_sequence_parallel_size': 1},
|
| 52 |
+
[36m(TaskRunner pid=1165231)[0m 'rollout': {'disable_log_stats': True,
|
| 53 |
+
[36m(TaskRunner pid=1165231)[0m 'do_sample': True,
|
| 54 |
+
[36m(TaskRunner pid=1165231)[0m 'dtype': 'bfloat16',
|
| 55 |
+
[36m(TaskRunner pid=1165231)[0m 'enable_chunked_prefill': False,
|
| 56 |
+
[36m(TaskRunner pid=1165231)[0m 'enforce_eager': False,
|
| 57 |
+
[36m(TaskRunner pid=1165231)[0m 'free_cache_engine': False,
|
| 58 |
+
[36m(TaskRunner pid=1165231)[0m 'gpu_memory_utilization': 0.4,
|
| 59 |
+
[36m(TaskRunner pid=1165231)[0m 'ignore_eos': False,
|
| 60 |
+
[36m(TaskRunner pid=1165231)[0m 'load_format': 'dummy_dtensor',
|
| 61 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 62 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_micro_batch_size': None,
|
| 63 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 64 |
+
[36m(TaskRunner pid=1165231)[0m 'log_prob_use_dynamic_bsz': False,
|
| 65 |
+
[36m(TaskRunner pid=1165231)[0m 'max_model_len': None,
|
| 66 |
+
[36m(TaskRunner pid=1165231)[0m 'max_num_batched_tokens': 8192,
|
| 67 |
+
[36m(TaskRunner pid=1165231)[0m 'max_num_seqs': 1024,
|
| 68 |
+
[36m(TaskRunner pid=1165231)[0m 'n': 5,
|
| 69 |
+
[36m(TaskRunner pid=1165231)[0m 'name': 'vllm',
|
| 70 |
+
[36m(TaskRunner pid=1165231)[0m 'prompt_length': 1024,
|
| 71 |
+
[36m(TaskRunner pid=1165231)[0m 'response_length': 1024,
|
| 72 |
+
[36m(TaskRunner pid=1165231)[0m 'temperature': 1.0,
|
| 73 |
+
[36m(TaskRunner pid=1165231)[0m 'tensor_model_parallel_size': 2,
|
| 74 |
+
[36m(TaskRunner pid=1165231)[0m 'top_k': -1,
|
| 75 |
+
[36m(TaskRunner pid=1165231)[0m 'top_p': 1,
|
| 76 |
+
[36m(TaskRunner pid=1165231)[0m 'use_fire_sampling': False,
|
| 77 |
+
[36m(TaskRunner pid=1165231)[0m 'val_kwargs': {'do_sample': False,
|
| 78 |
+
[36m(TaskRunner pid=1165231)[0m 'n': 1,
|
| 79 |
+
[36m(TaskRunner pid=1165231)[0m 'temperature': 0.4,
|
| 80 |
+
[36m(TaskRunner pid=1165231)[0m 'top_k': -1,
|
| 81 |
+
[36m(TaskRunner pid=1165231)[0m 'top_p': 1.0}}},
|
| 82 |
+
[36m(TaskRunner pid=1165231)[0m 'algorithm': {'adv_estimator': 'grpo',
|
| 83 |
+
[36m(TaskRunner pid=1165231)[0m 'gamma': 1.0,
|
| 84 |
+
[36m(TaskRunner pid=1165231)[0m 'kl_ctrl': {'kl_coef': 0.001, 'type': 'fixed'},
|
| 85 |
+
[36m(TaskRunner pid=1165231)[0m 'kl_penalty': 'kl',
|
| 86 |
+
[36m(TaskRunner pid=1165231)[0m 'lam': 1.0},
|
| 87 |
+
[36m(TaskRunner pid=1165231)[0m 'critic': {'cliprange_value': 0.5,
|
| 88 |
+
[36m(TaskRunner pid=1165231)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 89 |
+
[36m(TaskRunner pid=1165231)[0m 'forward_micro_batch_size': None,
|
| 90 |
+
[36m(TaskRunner pid=1165231)[0m 'forward_micro_batch_size_per_gpu': None,
|
| 91 |
+
[36m(TaskRunner pid=1165231)[0m 'grad_clip': 1.0,
|
| 92 |
+
[36m(TaskRunner pid=1165231)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 93 |
+
[36m(TaskRunner pid=1165231)[0m 'external_lib': None,
|
| 94 |
+
[36m(TaskRunner pid=1165231)[0m 'fsdp_config': {'fsdp_size': -1,[36m(TaskRunner pid=1165231)[0m DeprecationWarning: `ray.state.available_resources_per_node` is a private attribute and access will be removed in a future Ray version.
|
| 95 |
+
|
| 96 |
+
[36m(TaskRunner pid=1165231)[0m 'optimizer_offload': False,
|
| 97 |
+
[36m(TaskRunner pid=1165231)[0m 'param_offload': False,
|
| 98 |
+
[36m(TaskRunner pid=1165231)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 99 |
+
[36m(TaskRunner pid=1165231)[0m 'override_config': {},
|
| 100 |
+
[36m(TaskRunner pid=1165231)[0m 'path': '~/models/deepseek-llm-7b-chat',
|
| 101 |
+
[36m(TaskRunner pid=1165231)[0m 'tokenizer_path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 102 |
+
[36m(TaskRunner pid=1165231)[0m 'use_remove_padding': False},
|
| 103 |
+
[36m(TaskRunner pid=1165231)[0m 'optim': {'lr': 1e-05,
|
| 104 |
+
[36m(TaskRunner pid=1165231)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 105 |
+
[36m(TaskRunner pid=1165231)[0m 'min_lr_ratio': None,
|
| 106 |
+
[36m(TaskRunner pid=1165231)[0m 'total_training_steps': -1,
|
| 107 |
+
[36m(TaskRunner pid=1165231)[0m 'warmup_style': 'constant'},
|
| 108 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_epochs': 1,
|
| 109 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_max_token_len_per_gpu': 32768,
|
| 110 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_micro_batch_size': None,
|
| 111 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_micro_batch_size_per_gpu': None,
|
| 112 |
+
[36m(TaskRunner pid=1165231)[0m 'ppo_mini_batch_size': 16,
|
| 113 |
+
[36m(TaskRunner pid=1165231)[0m 'shuffle': False,
|
| 114 |
+
[36m(TaskRunner pid=1165231)[0m 'strategy': 'fsdp',
|
| 115 |
+
[36m(TaskRunner pid=1165231)[0m 'ulysses_sequence_parallel_size': 1,
|
| 116 |
+
[36m(TaskRunner pid=1165231)[0m 'use_dynamic_bsz': False},
|
| 117 |
+
[36m(TaskRunner pid=1165231)[0m 'custom_reward_function': {'name': 'compute_score', 'path': None},
|
| 118 |
+
[36m(TaskRunner pid=1165231)[0m 'data': {'filter_overlong_prompts': True,
|
| 119 |
+
[36m(TaskRunner pid=1165231)[0m 'image_key': 'images',
|
| 120 |
+
[36m(TaskRunner pid=1165231)[0m 'max_prompt_length': 1024,
|
| 121 |
+
[36m(TaskRunner pid=1165231)[0m 'max_response_length': 1024,
|
| 122 |
+
[36m(TaskRunner pid=1165231)[0m 'prompt_key': 'prompt',
|
| 123 |
+
[36m(TaskRunner pid=1165231)[0m 'return_raw_chat': False,
|
| 124 |
+
[36m(TaskRunner pid=1165231)[0m 'return_raw_input_ids': False,
|
| 125 |
+
[36m(TaskRunner pid=1165231)[0m 'shuffle': True,
|
| 126 |
+
[36m(TaskRunner pid=1165231)[0m 'tokenizer': None,
|
| 127 |
+
[36m(TaskRunner pid=1165231)[0m 'train_batch_size': 16,
|
| 128 |
+
[36m(TaskRunner pid=1165231)[0m 'train_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet',
|
| 129 |
+
[36m(TaskRunner pid=1165231)[0m 'truncation': 'error',
|
| 130 |
+
[36m(TaskRunner pid=1165231)[0m 'val_batch_size': 1,
|
| 131 |
+
[36m(TaskRunner pid=1165231)[0m 'val_files': '/tmp/anomseer_eval_smoke.parquet'},
|
| 132 |
+
[36m(TaskRunner pid=1165231)[0m 'reward_model': {'enable': False,
|
| 133 |
+
[36m(TaskRunner pid=1165231)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 134 |
+
[36m(TaskRunner pid=1165231)[0m 'max_length': None,
|
| 135 |
+
[36m(TaskRunner pid=1165231)[0m 'micro_batch_size': None,
|
| 136 |
+
[36m(TaskRunner pid=1165231)[0m 'micro_batch_size_per_gpu': None,
|
| 137 |
+
[36m(TaskRunner pid=1165231)[0m 'model': {'external_lib': None,
|
| 138 |
+
[36m(TaskRunner pid=1165231)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 139 |
+
[36m(TaskRunner pid=1165231)[0m 'param_offload': False,
|
| 140 |
+
[36m(TaskRunner pid=1165231)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 141 |
+
[36m(TaskRunner pid=1165231)[0m 'input_tokenizer': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 142 |
+
[36m(TaskRunner pid=1165231)[0m 'path': '~/models/FsfairX-LLaMA3-RM-v0.1',
|
| 143 |
+
[36m(TaskRunner pid=1165231)[0m 'use_remove_padding': False},
|
| 144 |
+
[36m(TaskRunner pid=1165231)[0m 'reward_manager': 'naive',
|
| 145 |
+
[36m(TaskRunner pid=1165231)[0m 'strategy': 'fsdp',
|
| 146 |
+
[36m(TaskRunner pid=1165231)[0m 'ulysses_sequence_parallel_size': 1,
|
| 147 |
+
[36m(TaskRunner pid=1165231)[0m 'use_dynamic_bsz': False},
|
| 148 |
+
[36m(TaskRunner pid=1165231)[0m 'trainer': {'balance_batch': True,
|
| 149 |
+
[36m(TaskRunner pid=1165231)[0m 'critic_warmup': 0,
|
| 150 |
+
[36m(TaskRunner pid=1165231)[0m 'default_hdfs_dir': None,
|
| 151 |
+
[36m(TaskRunner pid=1165231)[0m 'default_local_dir': 'checkpoints/anomseer/anomseer_rats_uni_2gpu',
|
| 152 |
+
[36m(TaskRunner pid=1165231)[0m 'del_local_ckpt_after_load': False,
|
| 153 |
+
[36m(TaskRunner pid=1165231)[0m 'experiment_name': 'anomseer_rats_uni_2gpu',
|
| 154 |
+
[36m(TaskRunner pid=1165231)[0m 'logger': ['console'],
|
| 155 |
+
[36m(TaskRunner pid=1165231)[0m 'n_gpus_per_node': 2,
|
| 156 |
+
[36m(TaskRunner pid=1165231)[0m 'nnodes': 1,
|
| 157 |
+
[36m(TaskRunner pid=1165231)[0m 'project_name': 'anomseer',
|
| 158 |
+
[36m(TaskRunner pid=1165231)[0m 'remove_previous_ckpt_in_save': False,
|
| 159 |
+
[36m(TaskRunner pid=1165231)[0m 'resume_from_path': False,
|
| 160 |
+
[36m(TaskRunner pid=1165231)[0m 'resume_mode': 'disable',
|
| 161 |
+
[36m(TaskRunner pid=1165231)[0m 'save_freq': 500,
|
| 162 |
+
[36m(TaskRunner pid=1165231)[0m 'test_freq': -1,
|
| 163 |
+
[36m(TaskRunner pid=1165231)[0m 'total_epochs': 1,
|
| 164 |
+
[36m(TaskRunner pid=1165231)[0m 'total_training_steps': None,
|
| 165 |
+
[36m(TaskRunner pid=1165231)[0m 'val_before_train': True,
|
| 166 |
+
[36m(TaskRunner pid=1165231)[0m 'val_generations_to_log_to_wandb': 0,
|
| 167 |
+
[36m(TaskRunner pid=1165231)[0m 'val_only': True},
|
| 168 |
+
[36m(TaskRunner pid=1165231)[0m 'ts': {'adv_mix': 0.3,
|
| 169 |
+
[36m(TaskRunner pid=1165231)[0m 'clip_pooling': 'mean',
|
| 170 |
+
[36m(TaskRunner pid=1165231)[0m 'clip_temperature': 1.0,
|
| 171 |
+
[36m(TaskRunner pid=1165231)[0m 'ot_eps': 0.08,
|
| 172 |
+
[36m(TaskRunner pid=1165231)[0m 'ot_n_iter': 50,
|
| 173 |
+
[36m(TaskRunner pid=1165231)[0m 'similarity_method': 'ot',
|
| 174 |
+
[36m(TaskRunner pid=1165231)[0m 'use_sem_orth': True}}
|
| 175 |
+
[36m(TaskRunner pid=1165231)[0m reward_manager_name: anomts
|
| 176 |
+
[36m(TaskRunner pid=1165231)[0m WARNING: val_batch_size is deprecated. Validation datasets are sent to inference engines as a whole batch, which will schedule the memory themselves.
|
| 177 |
+
[36m(TaskRunner pid=1165231)[0m [validate_config] All configuration checks passed successfully!
|
| 178 |
+
[36m(TaskRunner pid=1165231)[0m dataset len: 7566
|
| 179 |
+
[36m(TaskRunner pid=1165231)[0m filter dataset len: 7566
|
| 180 |
+
[36m(TaskRunner pid=1165231)[0m dataset len: 1
|
| 181 |
+
[36m(TaskRunner pid=1165231)[0m filter dataset len: 1
|
| 182 |
+
[36m(TaskRunner pid=1165231)[0m Size of train dataloader: 472
|
| 183 |
+
[36m(TaskRunner pid=1165231)[0m Total training steps: 472
|
| 184 |
+
[36m(WorkerDict pid=1168371)[0m Model config after override: Qwen2_5_VLConfig {
|
| 185 |
+
[36m(WorkerDict pid=1168371)[0m "architectures": [
|
| 186 |
+
[36m(WorkerDict pid=1168371)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 187 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 188 |
+
[36m(WorkerDict pid=1168371)[0m "attention_dropout": 0.0,
|
| 189 |
+
[36m(WorkerDict pid=1168371)[0m "eos_token_id": 151645,
|
| 190 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 191 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 2048,
|
| 192 |
+
[36m(WorkerDict pid=1168371)[0m "image_token_id": 151655,
|
| 193 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 194 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 11008,
|
| 195 |
+
[36m(WorkerDict pid=1168371)[0m "max_position_embeddings": 128000,
|
| 196 |
+
[36m(WorkerDict pid=1168371)[0m "max_window_layers": 70,
|
| 197 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl",
|
| 198 |
+
[36m(WorkerDict pid=1168371)[0m "num_attention_heads": 16,
|
| 199 |
+
[36m(WorkerDict pid=1168371)[0m "num_hidden_layers": 36,
|
| 200 |
+
[36m(WorkerDict pid=1168371)[0m "num_key_value_heads": 2,[36m(WorkerDict pid=1168371)[0m
|
| 201 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 202 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 203 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 204 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 205 |
+
[36m(WorkerDict pid=1168371)[0m [rank0]:[W615 11:19:02.122526228 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 0] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 206 |
+
|
| 207 |
+
[36m(WorkerDict pid=1168371)[0m "pad_token_id": 151643,
|
| 208 |
+
[36m(WorkerDict pid=1168371)[0m "rms_norm_eps": 1e-06,
|
| 209 |
+
[36m(WorkerDict pid=1168371)[0m "rope_scaling": {
|
| 210 |
+
[36m(WorkerDict pid=1168371)[0m "mrope_section": [
|
| 211 |
+
[36m(WorkerDict pid=1168371)[0m 16,
|
| 212 |
+
[36m(WorkerDict pid=1168371)[0m 24,
|
| 213 |
+
[36m(WorkerDict pid=1168371)[0m 24
|
| 214 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 215 |
+
[36m(WorkerDict pid=1168371)[0m "rope_type": "default",
|
| 216 |
+
[36m(WorkerDict pid=1168371)[0m "type": "default"
|
| 217 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 218 |
+
[36m(WorkerDict pid=1168371)[0m "rope_theta": 1000000.0,
|
| 219 |
+
[36m(WorkerDict pid=1168371)[0m "sliding_window": 32768,
|
| 220 |
+
[36m(WorkerDict pid=1168371)[0m "text_config": {
|
| 221 |
+
[36m(WorkerDict pid=1168371)[0m "architectures": [
|
| 222 |
+
[36m(WorkerDict pid=1168371)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 223 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 224 |
+
[36m(WorkerDict pid=1168371)[0m "attention_dropout": 0.0,
|
| 225 |
+
[36m(WorkerDict pid=1168371)[0m "bos_token_id": 151643,
|
| 226 |
+
[36m(WorkerDict pid=1168371)[0m "eos_token_id": 151645,
|
| 227 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 228 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 2048,
|
| 229 |
+
[36m(WorkerDict pid=1168371)[0m "image_token_id": null,
|
| 230 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 231 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 11008,
|
| 232 |
+
[36m(WorkerDict pid=1168371)[0m "layer_types": [
|
| 233 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 234 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 235 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 236 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 237 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 238 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 239 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 240 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 241 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 242 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 243 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 244 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 245 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 246 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 247 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 248 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 249 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 250 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 251 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 252 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 253 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 254 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 255 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 256 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 257 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 258 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 259 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 260 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 261 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 262 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 263 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 264 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 265 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 266 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 267 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 268 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention"
|
| 269 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 270 |
+
[36m(WorkerDict pid=1168371)[0m "max_position_embeddings": 128000,
|
| 271 |
+
[36m(WorkerDict pid=1168371)[0m "max_window_layers": 70,
|
| 272 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl_text",
|
| 273 |
+
[36m(WorkerDict pid=1168371)[0m "num_attention_heads": 16,
|
| 274 |
+
[36m(WorkerDict pid=1168371)[0m "num_hidden_layers": 36,
|
| 275 |
+
[36m(WorkerDict pid=1168371)[0m "num_key_value_heads": 2,
|
| 276 |
+
[36m(WorkerDict pid=1168371)[0m "rms_norm_eps": 1e-06,
|
| 277 |
+
[36m(WorkerDict pid=1168371)[0m "rope_scaling": {
|
| 278 |
+
[36m(WorkerDict pid=1168371)[0m "mrope_section": [
|
| 279 |
+
[36m(WorkerDict pid=1168371)[0m 16,
|
| 280 |
+
[36m(WorkerDict pid=1168371)[0m 24,
|
| 281 |
+
[36m(WorkerDict pid=1168371)[0m 24
|
| 282 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 283 |
+
[36m(WorkerDict pid=1168371)[0m "rope_type": "default",
|
| 284 |
+
[36m(WorkerDict pid=1168371)[0m "type": "default"
|
| 285 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 286 |
+
[36m(WorkerDict pid=1168371)[0m "rope_theta": 1000000.0,
|
| 287 |
+
[36m(WorkerDict pid=1168371)[0m "sliding_window": null,
|
| 288 |
+
[36m(WorkerDict pid=1168371)[0m "tie_word_embeddings": true,
|
| 289 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 290 |
+
[36m(WorkerDict pid=1168371)[0m "use_cache": true,
|
| 291 |
+
[36m(WorkerDict pid=1168371)[0m "use_sliding_window": false,
|
| 292 |
+
[36m(WorkerDict pid=1168371)[0m "video_token_id": null,
|
| 293 |
+
[36m(WorkerDict pid=1168371)[0m "vision_end_token_id": 151653,
|
| 294 |
+
[36m(WorkerDict pid=1168371)[0m "vision_start_token_id": 151652,
|
| 295 |
+
[36m(WorkerDict pid=1168371)[0m "vision_token_id": 151654,
|
| 296 |
+
[36m(WorkerDict pid=1168371)[0m "vocab_size": 151936
|
| 297 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 298 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 299 |
+
[36m(WorkerDict pid=1168371)[0m "transformers_version": "4.54.1",
|
| 300 |
+
[36m(WorkerDict pid=1168371)[0m "use_cache": true,
|
| 301 |
+
[36m(WorkerDict pid=1168371)[0m "use_sliding_window": false,
|
| 302 |
+
[36m(WorkerDict pid=1168371)[0m "video_token_id": 151656,
|
| 303 |
+
[36m(WorkerDict pid=1168371)[0m "vision_config": {
|
| 304 |
+
[36m(WorkerDict pid=1168371)[0m "depth": 32,
|
| 305 |
+
[36m(WorkerDict pid=1168371)[0m "fullatt_block_indexes": [
|
| 306 |
+
[36m(WorkerDict pid=1168371)[0m 7,
|
| 307 |
+
[36m(WorkerDict pid=1168371)[0m 15,
|
| 308 |
+
[36m(WorkerDict pid=1168371)[0m 23,
|
| 309 |
+
[36m(WorkerDict pid=1168371)[0m 31
|
| 310 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 311 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 312 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 1280,
|
| 313 |
+
[36m(WorkerDict pid=1168371)[0m "in_channels": 3,
|
| 314 |
+
[36m(WorkerDict pid=1168371)[0m "in_chans": 3,
|
| 315 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 316 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 3420,
|
| 317 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl",
|
| 318 |
+
[36m(WorkerDict pid=1168371)[0m "num_heads": 16,
|
| 319 |
+
[36m(WorkerDict pid=1168371)[0m "out_hidden_size": 2048,
|
| 320 |
+
[36m(WorkerDict pid=1168371)[0m "patch_size": 14,
|
| 321 |
+
[36m(WorkerDict pid=1168371)[0m "spatial_merge_size": 2,
|
| 322 |
+
[36m(WorkerDict pid=1168371)[0m "spatial_patch_size": 14,
|
| 323 |
+
[36m(WorkerDict pid=1168371)[0m "temporal_patch_size": 2,
|
| 324 |
+
[36m(WorkerDict pid=1168371)[0m "tokens_per_second": 2,
|
| 325 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 326 |
+
[36m(WorkerDict pid=1168371)[0m "window_size": 112
|
| 327 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 328 |
+
[36m(WorkerDict pid=1168371)[0m "vision_end_token_id": 151653,
|
| 329 |
+
[36m(WorkerDict pid=1168371)[0m "vision_start_token_id": 151652,
|
| 330 |
+
[36m(WorkerDict pid=1168371)[0m "vision_token_id": 151654,
|
| 331 |
+
[36m(WorkerDict pid=1168371)[0m "vocab_size": 151936
|
| 332 |
+
[36m(WorkerDict pid=1168371)[0m }
|
| 333 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 334 |
+
[36m(WorkerDict pid=1168371)[0m NCCL version 2.21.5+cuda12.4
|
| 335 |
+
[36m(WorkerDict pid=1168371)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 336 |
+
[36m(WorkerDict pid=1168371)[0m wrap_policy: functools.partial(<function _or_policy at 0x7af6274bc5e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7af6274bc4a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>})])
|
| 337 |
+
[36m(WorkerDict pid=1170483)[0m Actor use_remove_padding=False
|
| 338 |
+
[36m(WorkerDict pid=1170483)[0m wrap_policy: functools.partial(<function _or_policy at 0x78426bee45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x78426bee44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])[36m(WorkerDict pid=1168371)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLForConditionalGeneration is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 339 |
+
[36m(WorkerDict pid=1168371)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 340 |
+
[36m(WorkerDict pid=1168371)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VisionTransformerPretrainedModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 341 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 342 |
+
[36m(WorkerDict pid=1170483)[0m [rank1]:[W615 11:19:02.127933277 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 1] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 343 |
+
|
| 344 |
+
[36m(WorkerDict pid=1168371)[0m Model config after override: Qwen2_5_VLConfig {
|
| 345 |
+
[36m(WorkerDict pid=1168371)[0m "architectures": [
|
| 346 |
+
[36m(WorkerDict pid=1168371)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 347 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 348 |
+
[36m(WorkerDict pid=1168371)[0m "attention_dropout": 0.0,
|
| 349 |
+
[36m(WorkerDict pid=1168371)[0m "eos_token_id": 151645,
|
| 350 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 351 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 2048,
|
| 352 |
+
[36m(WorkerDict pid=1168371)[0m "image_token_id": 151655,
|
| 353 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 354 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 11008,
|
| 355 |
+
[36m(WorkerDict pid=1168371)[0m "max_position_embeddings": 128000,
|
| 356 |
+
[36m(WorkerDict pid=1168371)[0m "max_window_layers": 70,
|
| 357 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl",
|
| 358 |
+
[36m(WorkerDict pid=1168371)[0m "num_attention_heads": 16,
|
| 359 |
+
[36m(WorkerDict pid=1168371)[0m "num_hidden_layers": 36,
|
| 360 |
+
[36m(WorkerDict pid=1168371)[0m "num_key_value_heads": 2,
|
| 361 |
+
[36m(WorkerDict pid=1168371)[0m "pad_token_id": 151643,
|
| 362 |
+
[36m(WorkerDict pid=1168371)[0m "rms_norm_eps": 1e-06,
|
| 363 |
+
[36m(WorkerDict pid=1168371)[0m "rope_scaling": {
|
| 364 |
+
[36m(WorkerDict pid=1168371)[0m "mrope_section": [
|
| 365 |
+
[36m(WorkerDict pid=1168371)[0m 16,
|
| 366 |
+
[36m(WorkerDict pid=1168371)[0m 24,
|
| 367 |
+
[36m(WorkerDict pid=1168371)[0m 24
|
| 368 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 369 |
+
[36m(WorkerDict pid=1168371)[0m "rope_type": "default",
|
| 370 |
+
[36m(WorkerDict pid=1168371)[0m "type": "default"
|
| 371 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 372 |
+
[36m(WorkerDict pid=1168371)[0m "rope_theta": 1000000.0,
|
| 373 |
+
[36m(WorkerDict pid=1168371)[0m "sliding_window": 32768,
|
| 374 |
+
[36m(WorkerDict pid=1168371)[0m "text_config": {
|
| 375 |
+
[36m(WorkerDict pid=1168371)[0m "architectures": [
|
| 376 |
+
[36m(WorkerDict pid=1168371)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 377 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 378 |
+
[36m(WorkerDict pid=1168371)[0m "attention_dropout": 0.0,
|
| 379 |
+
[36m(WorkerDict pid=1168371)[0m "bos_token_id": 151643,
|
| 380 |
+
[36m(WorkerDict pid=1168371)[0m "eos_token_id": 151645,
|
| 381 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 382 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 2048,
|
| 383 |
+
[36m(WorkerDict pid=1168371)[0m "image_token_id": null,
|
| 384 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 385 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 11008,
|
| 386 |
+
[36m(WorkerDict pid=1168371)[0m "layer_types": [
|
| 387 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 388 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 389 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 390 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 391 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 392 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 393 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 394 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 395 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 396 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 397 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 398 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 399 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 400 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 401 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 402 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 403 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 404 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 405 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 406 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 407 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 408 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 409 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 410 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 411 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 412 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 413 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 414 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 415 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 416 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 417 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 418 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 419 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 420 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 421 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention",
|
| 422 |
+
[36m(WorkerDict pid=1168371)[0m "full_attention"
|
| 423 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 424 |
+
[36m(WorkerDict pid=1168371)[0m "max_position_embeddings": 128000,
|
| 425 |
+
[36m(WorkerDict pid=1168371)[0m "max_window_layers": 70,
|
| 426 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl_text",
|
| 427 |
+
[36m(WorkerDict pid=1168371)[0m "num_attention_heads": 16,
|
| 428 |
+
[36m(WorkerDict pid=1168371)[0m "num_hidden_layers": 36,
|
| 429 |
+
[36m(WorkerDict pid=1168371)[0m "num_key_value_heads": 2,
|
| 430 |
+
[36m(WorkerDict pid=1168371)[0m "rms_norm_eps": 1e-06,
|
| 431 |
+
[36m(WorkerDict pid=1168371)[0m "rope_scaling": {
|
| 432 |
+
[36m(WorkerDict pid=1168371)[0m "mrope_section": [
|
| 433 |
+
[36m(WorkerDict pid=1168371)[0m 16,
|
| 434 |
+
[36m(WorkerDict pid=1168371)[0m 24,
|
| 435 |
+
[36m(WorkerDict pid=1168371)[0m 24
|
| 436 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 437 |
+
[36m(WorkerDict pid=1168371)[0m "rope_type": "default",
|
| 438 |
+
[36m(WorkerDict pid=1168371)[0m "type": "default"
|
| 439 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 440 |
+
[36m(WorkerDict pid=1168371)[0m "rope_theta": 1000000.0,
|
| 441 |
+
[36m(WorkerDict pid=1168371)[0m "sliding_window": null,
|
| 442 |
+
[36m(WorkerDict pid=1168371)[0m "tie_word_embeddings": true,
|
| 443 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 444 |
+
[36m(WorkerDict pid=1168371)[0m "use_cache": true,
|
| 445 |
+
[36m(WorkerDict pid=1168371)[0m "use_sliding_window": false,
|
| 446 |
+
[36m(WorkerDict pid=1168371)[0m "video_token_id": null,
|
| 447 |
+
[36m(WorkerDict pid=1168371)[0m "vision_end_token_id": 151653,
|
| 448 |
+
[36m(WorkerDict pid=1168371)[0m "vision_start_token_id": 151652,
|
| 449 |
+
[36m(WorkerDict pid=1168371)[0m "vision_token_id": 151654,
|
| 450 |
+
[36m(WorkerDict pid=1168371)[0m "vocab_size": 151936
|
| 451 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 452 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 453 |
+
[36m(WorkerDict pid=1168371)[0m "transformers_version": "4.54.1",
|
| 454 |
+
[36m(WorkerDict pid=1168371)[0m "use_cache": true,
|
| 455 |
+
[36m(WorkerDict pid=1168371)[0m "use_sliding_window": false,
|
| 456 |
+
[36m(WorkerDict pid=1168371)[0m "video_token_id": 151656,
|
| 457 |
+
[36m(WorkerDict pid=1168371)[0m "vision_config": {
|
| 458 |
+
[36m(WorkerDict pid=1168371)[0m "depth": 32,
|
| 459 |
+
[36m(WorkerDict pid=1168371)[0m "fullatt_block_indexes": [
|
| 460 |
+
[36m(WorkerDict pid=1168371)[0m 7,
|
| 461 |
+
[36m(WorkerDict pid=1168371)[0m 15,
|
| 462 |
+
[36m(WorkerDict pid=1168371)[0m 23,
|
| 463 |
+
[36m(WorkerDict pid=1168371)[0m 31
|
| 464 |
+
[36m(WorkerDict pid=1168371)[0m ],
|
| 465 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_act": "silu",
|
| 466 |
+
[36m(WorkerDict pid=1168371)[0m "hidden_size": 1280,
|
| 467 |
+
[36m(WorkerDict pid=1168371)[0m "in_channels": 3,
|
| 468 |
+
[36m(WorkerDict pid=1168371)[0m "in_chans": 3,
|
| 469 |
+
[36m(WorkerDict pid=1168371)[0m "initializer_range": 0.02,
|
| 470 |
+
[36m(WorkerDict pid=1168371)[0m "intermediate_size": 3420,
|
| 471 |
+
[36m(WorkerDict pid=1168371)[0m "model_type": "qwen2_5_vl",
|
| 472 |
+
[36m(WorkerDict pid=1168371)[0m "num_heads": 16,
|
| 473 |
+
[36m(WorkerDict pid=1168371)[0m "out_hidden_size": 2048,
|
| 474 |
+
[36m(WorkerDict pid=1168371)[0m "patch_size": 14,
|
| 475 |
+
[36m(WorkerDict pid=1168371)[0m "spatial_merge_size": 2,
|
| 476 |
+
[36m(WorkerDict pid=1168371)[0m "spatial_patch_size": 14,
|
| 477 |
+
[36m(WorkerDict pid=1168371)[0m "temporal_patch_size": 2,
|
| 478 |
+
[36m(WorkerDict pid=1168371)[0m "tokens_per_second": 2,
|
| 479 |
+
[36m(WorkerDict pid=1168371)[0m "torch_dtype": "float32",
|
| 480 |
+
[36m(WorkerDict pid=1168371)[0m "window_size": 112
|
| 481 |
+
[36m(WorkerDict pid=1168371)[0m },
|
| 482 |
+
[36m(WorkerDict pid=1168371)[0m "vision_end_token_id": 151653,
|
| 483 |
+
[36m(WorkerDict pid=1168371)[0m "vision_start_token_id": 151652,
|
| 484 |
+
[36m(WorkerDict pid=1168371)[0m "vision_token_id": 151654,[36m(WorkerDict pid=1168371)[0m
|
| 485 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 486 |
+
[36m(WorkerDict pid=1170483)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`[32m [repeated 5x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/user-guides/configure-logging.html#log-deduplication for more options.)[0m
|
| 487 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 488 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 489 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 490 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 491 |
+
[36m(WorkerDict pid=1170483)[0m
|
| 492 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 493 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 494 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 495 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 496 |
+
[36m(WorkerDict pid=1168371)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 497 |
+
[36m(WorkerDict pid=1168371)[0m warnings.warn(
|
| 498 |
+
[36m(TaskRunner pid=1165231)[0m Unhandled error (suppress with 'RAY_IGNORE_UNHANDLED_ERRORS=1'): [36mray::WorkerDict.actor_rollout_generate_sequences()[39m (pid=1170483, ip=172.20.115.25, actor_id=33049b605f0612dfcfe2c38801000000, repr=<verl.single_controller.ray.base.WorkerDict object at 0x7842040bb090>)
|
| 499 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 500 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 501 |
+
[36m(TaskRunner pid=1165231)[0m File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 420, in func
|
| 502 |
+
[36m(TaskRunner pid=1165231)[0m return getattr(self.worker_dict[key], name)(*args, **kwargs)
|
| 503 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 504 |
+
[36m(TaskRunner pid=1165231)[0m File "/mnt/share01/sqk/AnomSeer/verl/single_controller/base/decorator.py", line 404, in inner
|
| 505 |
+
[36m(TaskRunner pid=1165231)[0m return func(*args, **kwargs)
|
| 506 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^
|
| 507 |
+
[36m(TaskRunner pid=1165231)[0m File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 552, in generate_sequences
|
| 508 |
+
[36m(TaskRunner pid=1165231)[0m with self.rollout_sharding_manager:
|
| 509 |
+
[36m(TaskRunner pid=1165231)[0m File "/mnt/share01/sqk/AnomSeer/verl/workers/sharding_manager/fsdp_vllm.py", line 134, in __enter__
|
| 510 |
+
[36m(TaskRunner pid=1165231)[0m loaded_params = model.load_weights(
|
| 511 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^
|
| 512 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2_5_vl.py", line 1126, in load_weights
|
| 513 |
+
[36m(TaskRunner pid=1165231)[0m return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
|
| 514 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 515 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 261, in load_weights
|
| 516 |
+
[36m(TaskRunner pid=1165231)[0m autoloaded_weights = set(self._load_module("", self.module, weights))
|
| 517 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 518 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 222, in _load_module
|
| 519 |
+
[36m(TaskRunner pid=1165231)[0m yield from self._load_module(prefix,
|
| 520 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 195, in _load_module
|
| 521 |
+
[36m(TaskRunner pid=1165231)[0m loaded_params = module_load_weights(weights)
|
| 522 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 523 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2.py", line 486, in load_weights
|
| 524 |
+
[36m(TaskRunner pid=1165231)[0m return loader.load_weights(weights)
|
| 525 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 526 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 261, in load_weights
|
| 527 |
+
[36m(TaskRunner pid=1165231)[0m autoloaded_weights = set(self._load_module("", self.module, weights))
|
| 528 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 529 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 222, in _load_module
|
| 530 |
+
[36m(TaskRunner pid=1165231)[0m yield from self._load_module(prefix,
|
| 531 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 195, in _load_module
|
| 532 |
+
[36m(TaskRunner pid=1165231)[0m loaded_params = module_load_weights(weights)
|
| 533 |
+
[36m(TaskRunner pid=1165231)[0m ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 534 |
+
[36m(TaskRunner pid=1165231)[0m File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2.py", line 405, in load_weights
|
| 535 |
+
[36m(TaskRunner pid=1165231)[0m param = params_dict[name]
|
| 536 |
+
[36m(TaskRunner pid=1165231)[0m ~~~~~~~~~~~^^^^^^
|
| 537 |
+
[36m(TaskRunner pid=1165231)[0m KeyError: 'visual.patch_embed.proj.weight'
|
| 538 |
+
|
| 539 |
+
[36m(WorkerDict pid=1168371)[0m "vocab_size": 151936
|
| 540 |
+
[36m(WorkerDict pid=1168371)[0m }
|
| 541 |
+
[36m(WorkerDict pid=1168371)[0m
|
| 542 |
+
[36m(WorkerDict pid=1168371)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 543 |
+
[36m(WorkerDict pid=1168371)[0m wrap_policy: functools.partial(<function _or_policy at 0x7af6274bc5e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7af6274bc4a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>})])
|
| 544 |
+
[36m(WorkerDict pid=1168371)[0m Actor use_remove_padding=False
|
| 545 |
+
[36m(WorkerDict pid=1168371)[0m Total steps: 472, num_warmup_steps: 0
|
| 546 |
+
[36m(WorkerDict pid=1168371)[0m Actor use_remove_padding=False
|
| 547 |
+
[36m(WorkerDict pid=1168371)[0m Before building vllm rollout, memory allocated (GB): 7.027174949645996, memory reserved (GB): 12.189453125
|
| 548 |
+
[36m(WorkerDict pid=1168371)[0m WARNING 06-15 11:21:42 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x7af2b75a5d90>
|
| 549 |
+
[36m(WorkerDict pid=1170483)[0m wrap_policy: functools.partial(<function _or_policy at 0x78426bee45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x78426bee44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 550 |
+
[36m(WorkerDict pid=1170483)[0m Total steps: 472, num_warmup_steps: 0
|
| 551 |
+
[36m(WorkerDict pid=1170483)[0m Actor use_remove_padding=False
|
| 552 |
+
[36m(WorkerDict pid=1168371)[0m WARNING 06-15 11:21:48 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 553 |
+
[36m(WorkerDict pid=1170483)[0m WARNING 06-15 11:21:42 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x783e7118be90>
|
| 554 |
+
[36m(WorkerDict pid=1168371)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 555 |
+
[36m(WorkerDict pid=1168371)[0m After building vllm rollout, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 556 |
+
[36m(WorkerDict pid=1168371)[0m After building sharding manager, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 557 |
+
[36m(WorkerDict pid=1170483)[0m WARNING 06-15 11:21:48 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 558 |
+
[36m(TaskRunner pid=1165231)[0m Using LocalLogger is deprecated. The constructor API will change
|
| 559 |
+
Error executing job with overrides: ['algorithm.adv_estimator=grpo', 'data.train_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet', 'data.val_files=/tmp/anomseer_eval_smoke.parquet', 'data.train_batch_size=16', 'data.max_prompt_length=1024', 'data.max_response_length=1024', 'data.filter_overlong_prompts=True', 'data.truncation=error', 'data.image_key=images', 'actor_rollout_ref.model.path=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface', 'actor_rollout_ref.model.lora_rank=0', 'actor_rollout_ref.model.lora_alpha=16', 'actor_rollout_ref.model.lora_dropout=0.0', "actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'", 'actor_rollout_ref.actor.optim.lr=1e-4', 'actor_rollout_ref.model.use_remove_padding=False', 'actor_rollout_ref.actor.ppo_mini_batch_size=16', 'actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2', 'actor_rollout_ref.actor.use_kl_loss=True', 'actor_rollout_ref.actor.kl_loss_coef=0.001', 'actor_rollout_ref.actor.kl_loss_type=low_var_kl', 'actor_rollout_ref.model.enable_gradient_checkpointing=True', 'actor_rollout_ref.actor.fsdp_config.param_offload=False', 'actor_rollout_ref.actor.fsdp_config.optimizer_offload=False', 'actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.rollout.tensor_model_parallel_size=2', 'actor_rollout_ref.rollout.name=vllm', 'actor_rollout_ref.rollout.gpu_memory_utilization=0.4', 'actor_rollout_ref.rollout.enable_chunked_prefill=False', 'actor_rollout_ref.rollout.enforce_eager=False', 'actor_rollout_ref.rollout.free_cache_engine=False', 'actor_rollout_ref.rollout.n=5', 'actor_rollout_ref.rollout.val_kwargs.do_sample=False', 'actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.ref.fsdp_config.param_offload=True', 'algorithm.kl_ctrl.kl_coef=0.001', 'trainer.critic_warmup=0', "trainer.logger=['console']", 'trainer.project_name=anomseer', 'trainer.experiment_name=anomseer_rats_uni_2gpu', 'trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu', 'trainer.n_gpus_per_node=2', 'trainer.nnodes=1', 'trainer.save_freq=500', 'trainer.test_freq=10', 'trainer.val_only=True', 'trainer.val_before_train=True', 'trainer.resume_mode=disable', 'trainer.total_epochs=1', 'ts.use_sem_orth=True', 'ts.adv_mix=0.3', 'ts.similarity_method=ot', 'ts.ot_eps=0.08', 'ts.ot_n_iter=50', 'data.val_batch_size=1', 'trainer.test_freq=-1']
|
| 560 |
+
Traceback (most recent call last):
|
| 561 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 68, in main
|
| 562 |
+
ray.get(runner.run.remote(config))
|
| 563 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/auto_init_hook.py", line 22, in auto_init_wrapper
|
| 564 |
+
return fn(*args, **kwargs)
|
| 565 |
+
^^^^^^^^^^^^^^^^^^^
|
| 566 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/client_mode_hook.py", line 107, in wrapper
|
| 567 |
+
return func(*args, **kwargs)
|
| 568 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 569 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 2980, in get
|
| 570 |
+
values, debugger_breakpoint = worker.get_objects(
|
| 571 |
+
^^^^^^^^^^^^^^^^^^^
|
| 572 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 1023, in get_objects
|
| 573 |
+
raise value.as_instanceof_cause()
|
| 574 |
+
ray.exceptions.RayTaskError(KeyError): [36mray::TaskRunner.run()[39m (pid=1165231, ip=172.20.115.25, actor_id=58caf0b1df3b7b7227854f3c01000000, repr=<main_ppo.TaskRunner object at 0x74f018438150>)
|
| 575 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 576 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 577 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 173, in run
|
| 578 |
+
trainer.fit()
|
| 579 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/ppo/ray_trainer.py", line 2160, in fit
|
| 580 |
+
val_metrics = self._validate()
|
| 581 |
+
^^^^^^^^^^^^^^^^
|
| 582 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/ppo/ray_trainer.py", line 1849, in _validate
|
| 583 |
+
test_output_gen_batch_padded = self.actor_rollout_wg.generate_sequences(test_gen_batch_padded)
|
| 584 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 585 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 43, in func
|
| 586 |
+
output = ray.get(output)
|
| 587 |
+
^^^^^^^^^^^^^^^
|
| 588 |
+
^^^^^^^^^^^^^^^^^^^
|
| 589 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 590 |
+
^^^^^^^^^^^^^^^^^^^
|
| 591 |
+
ray.exceptions.RayTaskError(KeyError): [36mray::WorkerDict.actor_rollout_generate_sequences()[39m (pid=1168371, ip=172.20.115.25, actor_id=ccb80723c0cfb29a26c02beb01000000, repr=<verl.single_controller.ray.base.WorkerDict object at 0x7af5bfb1a510>)
|
| 592 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 593 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 594 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 420, in func
|
| 595 |
+
return getattr(self.worker_dict[key], name)(*args, **kwargs)
|
| 596 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 597 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/base/decorator.py", line 404, in inner
|
| 598 |
+
return func(*args, **kwargs)
|
| 599 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 600 |
+
File "/mnt/share01/sqk/AnomSeer/verl/workers/fsdp_workers.py", line 552, in generate_sequences
|
| 601 |
+
with self.rollout_sharding_manager:
|
| 602 |
+
File "/mnt/share01/sqk/AnomSeer/verl/workers/sharding_manager/fsdp_vllm.py", line 134, in __enter__
|
| 603 |
+
loaded_params = model.load_weights(
|
| 604 |
+
^^^^^^^^^^^^^^^^^^^
|
| 605 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2_5_vl.py", line 1126, in load_weights
|
| 606 |
+
return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
|
| 607 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 608 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 261, in load_weights
|
| 609 |
+
autoloaded_weights = set(self._load_module("", self.module, weights))
|
| 610 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 611 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 222, in _load_module
|
| 612 |
+
yield from self._load_module(prefix,
|
| 613 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 195, in _load_module
|
| 614 |
+
loaded_params = module_load_weights(weights)
|
| 615 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 616 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2.py", line 486, in load_weights
|
| 617 |
+
return loader.load_weights(weights)
|
| 618 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 619 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 261, in load_weights
|
| 620 |
+
autoloaded_weights = set(self._load_module("", self.module, weights))
|
| 621 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 622 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 222, in _load_module
|
| 623 |
+
yield from self._load_module(prefix,
|
| 624 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/utils.py", line 195, in _load_module
|
| 625 |
+
loaded_params = module_load_weights(weights)
|
| 626 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 627 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/vllm/model_executor/models/qwen2.py", line 405, in load_weights
|
| 628 |
+
param = params_dict[name]
|
| 629 |
+
~~~~~~~~~~~^^^^^^
|
| 630 |
+
KeyError: 'visual.patch_embed.proj.weight'
|
| 631 |
+
|
| 632 |
+
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
|
| 633 |
+
[36m(WorkerDict pid=1170483)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 634 |
+
[36m(WorkerDict pid=1170483)[0m warnings.warn(
|
| 635 |
+
[36m(WorkerDict pid=1170483)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_v2_20260615.log
ADDED
|
@@ -0,0 +1,803 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[log] saving full output to: /mnt/share01/sqk/AnomSeer/logs/anomseer_rats_uni_full_20260612_184427_exp3/rats_eval_lora_smoke_fixed_v2_20260615.log
|
| 2 |
+
[config] tuning=LoRA (r=16, alpha=16, target=q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj) | lr=1e-4 | gpus=2 | tensor-parallel (TP=2)
|
| 3 |
+
[config] model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 4 |
+
[config] python=/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python | stage=eval
|
| 5 |
+
[stage] EVAL (model=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface, gpus=2, tp=2, lora_rank=0)
|
| 6 |
+
2026-06-15 11:25:40,665 INFO worker.py:2012 -- Started a local Ray instance.
|
| 7 |
+
/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py:2051: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
|
| 8 |
+
warnings.warn(
|
| 9 |
+
[36m(TaskRunner pid=1203084)[0m {'actor_rollout_ref': {'actor': {'clip_ratio': 0.2,
|
| 10 |
+
[36m(TaskRunner pid=1203084)[0m 'entropy_coeff': 0.001,
|
| 11 |
+
[36m(TaskRunner pid=1203084)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 12 |
+
[36m(TaskRunner pid=1203084)[0m 'optimizer_offload': False,
|
| 13 |
+
[36m(TaskRunner pid=1203084)[0m 'param_offload': False,
|
| 14 |
+
[36m(TaskRunner pid=1203084)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 15 |
+
[36m(TaskRunner pid=1203084)[0m 'grad_clip': 1.0,
|
| 16 |
+
[36m(TaskRunner pid=1203084)[0m 'kl_loss_coef': 0.001,
|
| 17 |
+
[36m(TaskRunner pid=1203084)[0m 'kl_loss_type': 'low_var_kl',
|
| 18 |
+
[36m(TaskRunner pid=1203084)[0m 'optim': {'lr': 0.0001,
|
| 19 |
+
[36m(TaskRunner pid=1203084)[0m 'lr_warmup_steps': -1,
|
| 20 |
+
[36m(TaskRunner pid=1203084)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 21 |
+
[36m(TaskRunner pid=1203084)[0m 'min_lr_ratio': None,
|
| 22 |
+
[36m(TaskRunner pid=1203084)[0m 'total_training_steps': -1,
|
| 23 |
+
[36m(TaskRunner pid=1203084)[0m 'warmup_style': 'constant'},
|
| 24 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_epochs': 1,
|
| 25 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_max_token_len_per_gpu': 16384,
|
| 26 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_micro_batch_size': None,
|
| 27 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_micro_batch_size_per_gpu': 2,
|
| 28 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_mini_batch_size': 16,
|
| 29 |
+
[36m(TaskRunner pid=1203084)[0m 'shuffle': False,
|
| 30 |
+
[36m(TaskRunner pid=1203084)[0m 'strategy': 'fsdp',
|
| 31 |
+
[36m(TaskRunner pid=1203084)[0m 'ulysses_sequence_parallel_size': 1,
|
| 32 |
+
[36m(TaskRunner pid=1203084)[0m 'use_dynamic_bsz': False,
|
| 33 |
+
[36m(TaskRunner pid=1203084)[0m 'use_kl_loss': True,
|
| 34 |
+
[36m(TaskRunner pid=1203084)[0m 'use_torch_compile': True},
|
| 35 |
+
[36m(TaskRunner pid=1203084)[0m 'hybrid_engine': True,
|
| 36 |
+
[36m(TaskRunner pid=1203084)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 37 |
+
[36m(TaskRunner pid=1203084)[0m 'external_lib': None,
|
| 38 |
+
[36m(TaskRunner pid=1203084)[0m 'lora_alpha': 16,
|
| 39 |
+
[36m(TaskRunner pid=1203084)[0m 'lora_dropout': 0.0,
|
| 40 |
+
[36m(TaskRunner pid=1203084)[0m 'lora_rank': 0,
|
| 41 |
+
[36m(TaskRunner pid=1203084)[0m 'lora_target_modules': 'q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',
|
| 42 |
+
[36m(TaskRunner pid=1203084)[0m 'override_config': {},
|
| 43 |
+
[36m(TaskRunner pid=1203084)[0m 'path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 44 |
+
[36m(TaskRunner pid=1203084)[0m 'use_remove_padding': False},
|
| 45 |
+
[36m(TaskRunner pid=1203084)[0m 'ref': {'fsdp_config': {'param_offload': True,
|
| 46 |
+
[36m(TaskRunner pid=1203084)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 47 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 48 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_micro_batch_size': None,
|
| 49 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 50 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_use_dynamic_bsz': False,
|
| 51 |
+
[36m(TaskRunner pid=1203084)[0m 'ulysses_sequence_parallel_size': 1},
|
| 52 |
+
[36m(TaskRunner pid=1203084)[0m 'rollout': {'disable_log_stats': True,
|
| 53 |
+
[36m(TaskRunner pid=1203084)[0m 'do_sample': True,
|
| 54 |
+
[36m(TaskRunner pid=1203084)[0m 'dtype': 'bfloat16',
|
| 55 |
+
[36m(TaskRunner pid=1203084)[0m 'enable_chunked_prefill': False,
|
| 56 |
+
[36m(TaskRunner pid=1203084)[0m 'enforce_eager': False,
|
| 57 |
+
[36m(TaskRunner pid=1203084)[0m 'free_cache_engine': False,
|
| 58 |
+
[36m(TaskRunner pid=1203084)[0m 'gpu_memory_utilization': 0.4,
|
| 59 |
+
[36m(TaskRunner pid=1203084)[0m 'ignore_eos': False,
|
| 60 |
+
[36m(TaskRunner pid=1203084)[0m 'load_format': 'dummy_dtensor',
|
| 61 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 62 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_micro_batch_size': None,
|
| 63 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 64 |
+
[36m(TaskRunner pid=1203084)[0m 'log_prob_use_dynamic_bsz': False,
|
| 65 |
+
[36m(TaskRunner pid=1203084)[0m 'max_model_len': None,
|
| 66 |
+
[36m(TaskRunner pid=1203084)[0m 'max_num_batched_tokens': 8192,
|
| 67 |
+
[36m(TaskRunner pid=1203084)[0m 'max_num_seqs': 1024,
|
| 68 |
+
[36m(TaskRunner pid=1203084)[0m 'n': 5,
|
| 69 |
+
[36m(TaskRunner pid=1203084)[0m 'name': 'vllm',
|
| 70 |
+
[36m(TaskRunner pid=1203084)[0m 'prompt_length': 1024,
|
| 71 |
+
[36m(TaskRunner pid=1203084)[0m 'response_length': 1024,
|
| 72 |
+
[36m(TaskRunner pid=1203084)[0m 'temperature': 1.0,
|
| 73 |
+
[36m(TaskRunner pid=1203084)[0m 'tensor_model_parallel_size': 2,
|
| 74 |
+
[36m(TaskRunner pid=1203084)[0m 'top_k': -1,
|
| 75 |
+
[36m(TaskRunner pid=1203084)[0m 'top_p': 1,
|
| 76 |
+
[36m(TaskRunner pid=1203084)[0m 'use_fire_sampling': False,
|
| 77 |
+
[36m(TaskRunner pid=1203084)[0m 'val_kwargs': {'do_sample': False,
|
| 78 |
+
[36m(TaskRunner pid=1203084)[0m 'n': 1,
|
| 79 |
+
[36m(TaskRunner pid=1203084)[0m 'temperature': 0.4,
|
| 80 |
+
[36m(TaskRunner pid=1203084)[0m 'top_k': -1,
|
| 81 |
+
[36m(TaskRunner pid=1203084)[0m 'top_p': 1.0}}},
|
| 82 |
+
[36m(TaskRunner pid=1203084)[0m 'algorithm': {'adv_estimator': 'grpo',
|
| 83 |
+
[36m(TaskRunner pid=1203084)[0m 'gamma': 1.0,
|
| 84 |
+
[36m(TaskRunner pid=1203084)[0m 'kl_ctrl': {'kl_coef': 0.001, 'type': 'fixed'},
|
| 85 |
+
[36m(TaskRunner pid=1203084)[0m 'kl_penalty': 'kl',
|
| 86 |
+
[36m(TaskRunner pid=1203084)[0m 'lam': 1.0},
|
| 87 |
+
[36m(TaskRunner pid=1203084)[0m 'critic': {'cliprange_value': 0.5,
|
| 88 |
+
[36m(TaskRunner pid=1203084)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 89 |
+
[36m(TaskRunner pid=1203084)[0m 'forward_micro_batch_size': None,
|
| 90 |
+
[36m(TaskRunner pid=1203084)[0m 'forward_micro_batch_size_per_gpu': None,
|
| 91 |
+
[36m(TaskRunner pid=1203084)[0m 'grad_clip': 1.0,
|
| 92 |
+
[36m(TaskRunner pid=1203084)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 93 |
+
[36m(TaskRunner pid=1203084)[0m 'external_lib': None,
|
| 94 |
+
[36m(TaskRunner pid=1203084)[0m 'fsdp_config': {'fsdp_size': -1,[36m(TaskRunner pid=1203084)[0m DeprecationWarning: `ray.state.available_resources_per_node` is a private attribute and access will be removed in a future Ray version.
|
| 95 |
+
|
| 96 |
+
[36m(TaskRunner pid=1203084)[0m 'optimizer_offload': False,
|
| 97 |
+
[36m(TaskRunner pid=1203084)[0m 'param_offload': False,
|
| 98 |
+
[36m(TaskRunner pid=1203084)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 99 |
+
[36m(TaskRunner pid=1203084)[0m 'override_config': {},
|
| 100 |
+
[36m(TaskRunner pid=1203084)[0m 'path': '~/models/deepseek-llm-7b-chat',
|
| 101 |
+
[36m(TaskRunner pid=1203084)[0m 'tokenizer_path': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 102 |
+
[36m(TaskRunner pid=1203084)[0m 'use_remove_padding': False},
|
| 103 |
+
[36m(TaskRunner pid=1203084)[0m 'optim': {'lr': 1e-05,
|
| 104 |
+
[36m(TaskRunner pid=1203084)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 105 |
+
[36m(TaskRunner pid=1203084)[0m 'min_lr_ratio': None,
|
| 106 |
+
[36m(TaskRunner pid=1203084)[0m 'total_training_steps': -1,
|
| 107 |
+
[36m(TaskRunner pid=1203084)[0m 'warmup_style': 'constant'},
|
| 108 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_epochs': 1,
|
| 109 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_max_token_len_per_gpu': 32768,
|
| 110 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_micro_batch_size': None,
|
| 111 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_micro_batch_size_per_gpu': None,
|
| 112 |
+
[36m(TaskRunner pid=1203084)[0m 'ppo_mini_batch_size': 16,
|
| 113 |
+
[36m(TaskRunner pid=1203084)[0m 'shuffle': False,
|
| 114 |
+
[36m(TaskRunner pid=1203084)[0m 'strategy': 'fsdp',
|
| 115 |
+
[36m(TaskRunner pid=1203084)[0m 'ulysses_sequence_parallel_size': 1,
|
| 116 |
+
[36m(TaskRunner pid=1203084)[0m 'use_dynamic_bsz': False},
|
| 117 |
+
[36m(TaskRunner pid=1203084)[0m 'custom_reward_function': {'name': 'compute_score', 'path': None},
|
| 118 |
+
[36m(TaskRunner pid=1203084)[0m 'data': {'filter_overlong_prompts': True,
|
| 119 |
+
[36m(TaskRunner pid=1203084)[0m 'image_key': 'images',
|
| 120 |
+
[36m(TaskRunner pid=1203084)[0m 'max_prompt_length': 1024,
|
| 121 |
+
[36m(TaskRunner pid=1203084)[0m 'max_response_length': 1024,
|
| 122 |
+
[36m(TaskRunner pid=1203084)[0m 'prompt_key': 'prompt',
|
| 123 |
+
[36m(TaskRunner pid=1203084)[0m 'return_raw_chat': False,
|
| 124 |
+
[36m(TaskRunner pid=1203084)[0m 'return_raw_input_ids': False,
|
| 125 |
+
[36m(TaskRunner pid=1203084)[0m 'shuffle': True,
|
| 126 |
+
[36m(TaskRunner pid=1203084)[0m 'tokenizer': None,
|
| 127 |
+
[36m(TaskRunner pid=1203084)[0m 'train_batch_size': 16,
|
| 128 |
+
[36m(TaskRunner pid=1203084)[0m 'train_files': '/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet',
|
| 129 |
+
[36m(TaskRunner pid=1203084)[0m 'truncation': 'error',
|
| 130 |
+
[36m(TaskRunner pid=1203084)[0m 'val_batch_size': 1,
|
| 131 |
+
[36m(TaskRunner pid=1203084)[0m 'val_files': '/tmp/anomseer_eval_smoke.parquet'},
|
| 132 |
+
[36m(TaskRunner pid=1203084)[0m 'reward_model': {'enable': False,
|
| 133 |
+
[36m(TaskRunner pid=1203084)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 134 |
+
[36m(TaskRunner pid=1203084)[0m 'max_length': None,
|
| 135 |
+
[36m(TaskRunner pid=1203084)[0m 'micro_batch_size': None,
|
| 136 |
+
[36m(TaskRunner pid=1203084)[0m 'micro_batch_size_per_gpu': None,
|
| 137 |
+
[36m(TaskRunner pid=1203084)[0m 'model': {'external_lib': None,
|
| 138 |
+
[36m(TaskRunner pid=1203084)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 139 |
+
[36m(TaskRunner pid=1203084)[0m 'param_offload': False,
|
| 140 |
+
[36m(TaskRunner pid=1203084)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 141 |
+
[36m(TaskRunner pid=1203084)[0m 'input_tokenizer': '/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface',
|
| 142 |
+
[36m(TaskRunner pid=1203084)[0m 'path': '~/models/FsfairX-LLaMA3-RM-v0.1',
|
| 143 |
+
[36m(TaskRunner pid=1203084)[0m 'use_remove_padding': False},
|
| 144 |
+
[36m(TaskRunner pid=1203084)[0m 'reward_manager': 'naive',
|
| 145 |
+
[36m(TaskRunner pid=1203084)[0m 'strategy': 'fsdp',
|
| 146 |
+
[36m(TaskRunner pid=1203084)[0m 'ulysses_sequence_parallel_size': 1,
|
| 147 |
+
[36m(TaskRunner pid=1203084)[0m 'use_dynamic_bsz': False},
|
| 148 |
+
[36m(TaskRunner pid=1203084)[0m 'trainer': {'balance_batch': True,
|
| 149 |
+
[36m(TaskRunner pid=1203084)[0m 'critic_warmup': 0,
|
| 150 |
+
[36m(TaskRunner pid=1203084)[0m 'default_hdfs_dir': None,
|
| 151 |
+
[36m(TaskRunner pid=1203084)[0m 'default_local_dir': 'checkpoints/anomseer/anomseer_rats_uni_2gpu',
|
| 152 |
+
[36m(TaskRunner pid=1203084)[0m 'del_local_ckpt_after_load': False,
|
| 153 |
+
[36m(TaskRunner pid=1203084)[0m 'experiment_name': 'anomseer_rats_uni_2gpu',
|
| 154 |
+
[36m(TaskRunner pid=1203084)[0m 'logger': ['console'],
|
| 155 |
+
[36m(TaskRunner pid=1203084)[0m 'n_gpus_per_node': 2,
|
| 156 |
+
[36m(TaskRunner pid=1203084)[0m 'nnodes': 1,
|
| 157 |
+
[36m(TaskRunner pid=1203084)[0m 'project_name': 'anomseer',
|
| 158 |
+
[36m(TaskRunner pid=1203084)[0m 'remove_previous_ckpt_in_save': False,
|
| 159 |
+
[36m(TaskRunner pid=1203084)[0m 'resume_from_path': False,
|
| 160 |
+
[36m(TaskRunner pid=1203084)[0m 'resume_mode': 'disable',
|
| 161 |
+
[36m(TaskRunner pid=1203084)[0m 'save_freq': 500,
|
| 162 |
+
[36m(TaskRunner pid=1203084)[0m 'test_freq': -1,
|
| 163 |
+
[36m(TaskRunner pid=1203084)[0m 'total_epochs': 1,
|
| 164 |
+
[36m(TaskRunner pid=1203084)[0m 'total_training_steps': None,
|
| 165 |
+
[36m(TaskRunner pid=1203084)[0m 'val_before_train': True,
|
| 166 |
+
[36m(TaskRunner pid=1203084)[0m 'val_generations_to_log_to_wandb': 0,
|
| 167 |
+
[36m(TaskRunner pid=1203084)[0m 'val_only': True},
|
| 168 |
+
[36m(TaskRunner pid=1203084)[0m 'ts': {'adv_mix': 0.3,
|
| 169 |
+
[36m(TaskRunner pid=1203084)[0m 'clip_pooling': 'mean',
|
| 170 |
+
[36m(TaskRunner pid=1203084)[0m 'clip_temperature': 1.0,
|
| 171 |
+
[36m(TaskRunner pid=1203084)[0m 'ot_eps': 0.08,
|
| 172 |
+
[36m(TaskRunner pid=1203084)[0m 'ot_n_iter': 50,
|
| 173 |
+
[36m(TaskRunner pid=1203084)[0m 'similarity_method': 'ot',
|
| 174 |
+
[36m(TaskRunner pid=1203084)[0m 'use_sem_orth': True}}
|
| 175 |
+
[36m(TaskRunner pid=1203084)[0m reward_manager_name: anomts
|
| 176 |
+
[36m(TaskRunner pid=1203084)[0m WARNING: val_batch_size is deprecated. Validation datasets are sent to inference engines as a whole batch, which will schedule the memory themselves.
|
| 177 |
+
[36m(TaskRunner pid=1203084)[0m [validate_config] All configuration checks passed successfully!
|
| 178 |
+
[36m(TaskRunner pid=1203084)[0m dataset len: 7566
|
| 179 |
+
[36m(TaskRunner pid=1203084)[0m filter dataset len: 7566
|
| 180 |
+
[36m(TaskRunner pid=1203084)[0m dataset len: 1
|
| 181 |
+
[36m(TaskRunner pid=1203084)[0m filter dataset len: 1
|
| 182 |
+
[36m(TaskRunner pid=1203084)[0m Size of train dataloader: 472
|
| 183 |
+
[36m(TaskRunner pid=1203084)[0m Total training steps: 472
|
| 184 |
+
[36m(WorkerDict pid=1206072)[0m Model config after override: Qwen2_5_VLConfig {
|
| 185 |
+
[36m(WorkerDict pid=1206072)[0m "architectures": [
|
| 186 |
+
[36m(WorkerDict pid=1206072)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 187 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 188 |
+
[36m(WorkerDict pid=1206072)[0m "attention_dropout": 0.0,
|
| 189 |
+
[36m(WorkerDict pid=1206072)[0m "eos_token_id": 151645,
|
| 190 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 191 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 2048,
|
| 192 |
+
[36m(WorkerDict pid=1206072)[0m "image_token_id": 151655,
|
| 193 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 194 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 11008,
|
| 195 |
+
[36m(WorkerDict pid=1206072)[0m "max_position_embeddings": 128000,
|
| 196 |
+
[36m(WorkerDict pid=1206072)[0m "max_window_layers": 70,
|
| 197 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl",
|
| 198 |
+
[36m(WorkerDict pid=1206072)[0m "num_attention_heads": 16,
|
| 199 |
+
[36m(WorkerDict pid=1206072)[0m "num_hidden_layers": 36,
|
| 200 |
+
[36m(WorkerDict pid=1206072)[0m "num_key_value_heads": 2,[36m(WorkerDict pid=1208116)[0m
|
| 201 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 202 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 203 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 204 |
+
[36m(WorkerDict pid=1208116)[0m
|
| 205 |
+
[36m(WorkerDict pid=1206072)[0m [rank0]:[W615 11:26:41.949330793 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 0] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 206 |
+
|
| 207 |
+
[36m(WorkerDict pid=1206072)[0m "pad_token_id": 151643,
|
| 208 |
+
[36m(WorkerDict pid=1206072)[0m "rms_norm_eps": 1e-06,
|
| 209 |
+
[36m(WorkerDict pid=1206072)[0m "rope_scaling": {
|
| 210 |
+
[36m(WorkerDict pid=1206072)[0m "mrope_section": [
|
| 211 |
+
[36m(WorkerDict pid=1206072)[0m 16,
|
| 212 |
+
[36m(WorkerDict pid=1206072)[0m 24,
|
| 213 |
+
[36m(WorkerDict pid=1206072)[0m 24
|
| 214 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 215 |
+
[36m(WorkerDict pid=1206072)[0m "rope_type": "default",
|
| 216 |
+
[36m(WorkerDict pid=1206072)[0m "type": "default"
|
| 217 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 218 |
+
[36m(WorkerDict pid=1206072)[0m "rope_theta": 1000000.0,
|
| 219 |
+
[36m(WorkerDict pid=1206072)[0m "sliding_window": 32768,
|
| 220 |
+
[36m(WorkerDict pid=1206072)[0m "text_config": {
|
| 221 |
+
[36m(WorkerDict pid=1206072)[0m "architectures": [
|
| 222 |
+
[36m(WorkerDict pid=1206072)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 223 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 224 |
+
[36m(WorkerDict pid=1206072)[0m "attention_dropout": 0.0,
|
| 225 |
+
[36m(WorkerDict pid=1206072)[0m "bos_token_id": 151643,
|
| 226 |
+
[36m(WorkerDict pid=1206072)[0m "eos_token_id": 151645,
|
| 227 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 228 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 2048,
|
| 229 |
+
[36m(WorkerDict pid=1206072)[0m "image_token_id": null,
|
| 230 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 231 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 11008,
|
| 232 |
+
[36m(WorkerDict pid=1206072)[0m "layer_types": [
|
| 233 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 234 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 235 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 236 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 237 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 238 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 239 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 240 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 241 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 242 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 243 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 244 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 245 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 246 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 247 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 248 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 249 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 250 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 251 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 252 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 253 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 254 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 255 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 256 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 257 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 258 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 259 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 260 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 261 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 262 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 263 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 264 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 265 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 266 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 267 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 268 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention"
|
| 269 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 270 |
+
[36m(WorkerDict pid=1206072)[0m "max_position_embeddings": 128000,
|
| 271 |
+
[36m(WorkerDict pid=1206072)[0m "max_window_layers": 70,
|
| 272 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl_text",
|
| 273 |
+
[36m(WorkerDict pid=1206072)[0m "num_attention_heads": 16,
|
| 274 |
+
[36m(WorkerDict pid=1206072)[0m "num_hidden_layers": 36,
|
| 275 |
+
[36m(WorkerDict pid=1206072)[0m "num_key_value_heads": 2,
|
| 276 |
+
[36m(WorkerDict pid=1206072)[0m "rms_norm_eps": 1e-06,
|
| 277 |
+
[36m(WorkerDict pid=1206072)[0m "rope_scaling": {
|
| 278 |
+
[36m(WorkerDict pid=1206072)[0m "mrope_section": [
|
| 279 |
+
[36m(WorkerDict pid=1206072)[0m 16,
|
| 280 |
+
[36m(WorkerDict pid=1206072)[0m 24,
|
| 281 |
+
[36m(WorkerDict pid=1206072)[0m 24
|
| 282 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 283 |
+
[36m(WorkerDict pid=1206072)[0m "rope_type": "default",
|
| 284 |
+
[36m(WorkerDict pid=1206072)[0m "type": "default"
|
| 285 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 286 |
+
[36m(WorkerDict pid=1206072)[0m "rope_theta": 1000000.0,
|
| 287 |
+
[36m(WorkerDict pid=1206072)[0m "sliding_window": null,
|
| 288 |
+
[36m(WorkerDict pid=1206072)[0m "tie_word_embeddings": true,
|
| 289 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 290 |
+
[36m(WorkerDict pid=1206072)[0m "use_cache": true,
|
| 291 |
+
[36m(WorkerDict pid=1206072)[0m "use_sliding_window": false,
|
| 292 |
+
[36m(WorkerDict pid=1206072)[0m "video_token_id": null,
|
| 293 |
+
[36m(WorkerDict pid=1206072)[0m "vision_end_token_id": 151653,
|
| 294 |
+
[36m(WorkerDict pid=1206072)[0m "vision_start_token_id": 151652,
|
| 295 |
+
[36m(WorkerDict pid=1206072)[0m "vision_token_id": 151654,
|
| 296 |
+
[36m(WorkerDict pid=1206072)[0m "vocab_size": 151936
|
| 297 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 298 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 299 |
+
[36m(WorkerDict pid=1206072)[0m "transformers_version": "4.54.1",
|
| 300 |
+
[36m(WorkerDict pid=1206072)[0m "use_cache": true,
|
| 301 |
+
[36m(WorkerDict pid=1206072)[0m "use_sliding_window": false,
|
| 302 |
+
[36m(WorkerDict pid=1206072)[0m "video_token_id": 151656,
|
| 303 |
+
[36m(WorkerDict pid=1206072)[0m "vision_config": {
|
| 304 |
+
[36m(WorkerDict pid=1206072)[0m "depth": 32,
|
| 305 |
+
[36m(WorkerDict pid=1206072)[0m "fullatt_block_indexes": [
|
| 306 |
+
[36m(WorkerDict pid=1206072)[0m 7,
|
| 307 |
+
[36m(WorkerDict pid=1206072)[0m 15,
|
| 308 |
+
[36m(WorkerDict pid=1206072)[0m 23,
|
| 309 |
+
[36m(WorkerDict pid=1206072)[0m 31
|
| 310 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 311 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 312 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 1280,
|
| 313 |
+
[36m(WorkerDict pid=1206072)[0m "in_channels": 3,
|
| 314 |
+
[36m(WorkerDict pid=1206072)[0m "in_chans": 3,
|
| 315 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 316 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 3420,
|
| 317 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl",
|
| 318 |
+
[36m(WorkerDict pid=1206072)[0m "num_heads": 16,
|
| 319 |
+
[36m(WorkerDict pid=1206072)[0m "out_hidden_size": 2048,
|
| 320 |
+
[36m(WorkerDict pid=1206072)[0m "patch_size": 14,
|
| 321 |
+
[36m(WorkerDict pid=1206072)[0m "spatial_merge_size": 2,
|
| 322 |
+
[36m(WorkerDict pid=1206072)[0m "spatial_patch_size": 14,
|
| 323 |
+
[36m(WorkerDict pid=1206072)[0m "temporal_patch_size": 2,
|
| 324 |
+
[36m(WorkerDict pid=1206072)[0m "tokens_per_second": 2,
|
| 325 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 326 |
+
[36m(WorkerDict pid=1206072)[0m "window_size": 112
|
| 327 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 328 |
+
[36m(WorkerDict pid=1206072)[0m "vision_end_token_id": 151653,
|
| 329 |
+
[36m(WorkerDict pid=1206072)[0m "vision_start_token_id": 151652,
|
| 330 |
+
[36m(WorkerDict pid=1206072)[0m "vision_token_id": 151654,
|
| 331 |
+
[36m(WorkerDict pid=1206072)[0m "vocab_size": 151936
|
| 332 |
+
[36m(WorkerDict pid=1206072)[0m }
|
| 333 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 334 |
+
[36m(WorkerDict pid=1206072)[0m NCCL version 2.21.5+cuda12.4
|
| 335 |
+
[36m(WorkerDict pid=1206072)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 336 |
+
[36m(WorkerDict pid=1206072)[0m wrap_policy: functools.partial(<function _or_policy at 0x7cebba2d45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7cebba2d44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 337 |
+
[36m(WorkerDict pid=1206072)[0m Actor use_remove_padding=False
|
| 338 |
+
[36m(WorkerDict pid=1208116)[0m wrap_policy: functools.partial(<function _or_policy at 0x79a367cd45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x79a367cd44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 339 |
+
[36m(WorkerDict pid=1206072)[0m Model config after override: Qwen2_5_VLConfig {
|
| 340 |
+
[36m(WorkerDict pid=1206072)[0m "architectures": [
|
| 341 |
+
[36m(WorkerDict pid=1206072)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 342 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 343 |
+
[36m(WorkerDict pid=1206072)[0m "attention_dropout": 0.0,
|
| 344 |
+
[36m(WorkerDict pid=1206072)[0m "eos_token_id": 151645,
|
| 345 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 346 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 2048,
|
| 347 |
+
[36m(WorkerDict pid=1206072)[0m "image_token_id": 151655,
|
| 348 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 349 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 11008,
|
| 350 |
+
[36m(WorkerDict pid=1206072)[0m "max_position_embeddings": 128000,
|
| 351 |
+
[36m(WorkerDict pid=1206072)[0m "max_window_layers": 70,
|
| 352 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl",
|
| 353 |
+
[36m(WorkerDict pid=1206072)[0m "num_attention_heads": 16,
|
| 354 |
+
[36m(WorkerDict pid=1206072)[0m "num_hidden_layers": 36,
|
| 355 |
+
[36m(WorkerDict pid=1206072)[0m "num_key_value_heads": 2,
|
| 356 |
+
[36m(WorkerDict pid=1206072)[0m "pad_token_id": 151643,
|
| 357 |
+
[36m(WorkerDict pid=1206072)[0m "rms_norm_eps": 1e-06,
|
| 358 |
+
[36m(WorkerDict pid=1206072)[0m "rope_scaling": {
|
| 359 |
+
[36m(WorkerDict pid=1206072)[0m "mrope_section": [
|
| 360 |
+
[36m(WorkerDict pid=1206072)[0m 16,
|
| 361 |
+
[36m(WorkerDict pid=1206072)[0m 24,
|
| 362 |
+
[36m(WorkerDict pid=1206072)[0m 24
|
| 363 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 364 |
+
[36m(WorkerDict pid=1206072)[0m "rope_type": "default",
|
| 365 |
+
[36m(WorkerDict pid=1206072)[0m "type": "default"
|
| 366 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 367 |
+
[36m(WorkerDict pid=1206072)[0m "rope_theta": 1000000.0,
|
| 368 |
+
[36m(WorkerDict pid=1206072)[0m "sliding_window": 32768,
|
| 369 |
+
[36m(WorkerDict pid=1206072)[0m "text_config": {
|
| 370 |
+
[36m(WorkerDict pid=1206072)[0m "architectures": [
|
| 371 |
+
[36m(WorkerDict pid=1206072)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 372 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 373 |
+
[36m(WorkerDict pid=1206072)[0m "attention_dropout": 0.0,
|
| 374 |
+
[36m(WorkerDict pid=1206072)[0m "bos_token_id": 151643,
|
| 375 |
+
[36m(WorkerDict pid=1206072)[0m "eos_token_id": 151645,
|
| 376 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 377 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 2048,
|
| 378 |
+
[36m(WorkerDict pid=1206072)[0m "image_token_id": null,
|
| 379 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 380 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 11008,
|
| 381 |
+
[36m(WorkerDict pid=1206072)[0m "layer_types": [
|
| 382 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 383 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 384 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 385 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 386 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 387 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 388 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 389 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 390 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 391 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 392 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 393 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 394 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 395 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 396 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 397 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 398 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 399 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 400 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 401 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 402 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 403 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 404 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 405 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 406 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 407 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 408 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 409 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 410 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 411 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 412 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 413 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 414 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 415 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 416 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention",
|
| 417 |
+
[36m(WorkerDict pid=1206072)[0m "full_attention"
|
| 418 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 419 |
+
[36m(WorkerDict pid=1206072)[0m "max_position_embeddings": 128000,
|
| 420 |
+
[36m(WorkerDict pid=1206072)[0m "max_window_layers": 70,
|
| 421 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl_text",
|
| 422 |
+
[36m(WorkerDict pid=1206072)[0m "num_attention_heads": 16,
|
| 423 |
+
[36m(WorkerDict pid=1206072)[0m "num_hidden_layers": 36,
|
| 424 |
+
[36m(WorkerDict pid=1206072)[0m "num_key_value_heads": 2,
|
| 425 |
+
[36m(WorkerDict pid=1206072)[0m "rms_norm_eps": 1e-06,
|
| 426 |
+
[36m(WorkerDict pid=1206072)[0m "rope_scaling": {
|
| 427 |
+
[36m(WorkerDict pid=1206072)[0m "mrope_section": [
|
| 428 |
+
[36m(WorkerDict pid=1206072)[0m 16,
|
| 429 |
+
[36m(WorkerDict pid=1206072)[0m 24,
|
| 430 |
+
[36m(WorkerDict pid=1206072)[0m 24
|
| 431 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 432 |
+
[36m(WorkerDict pid=1206072)[0m "rope_type": "default",
|
| 433 |
+
[36m(WorkerDict pid=1206072)[0m "type": "default"
|
| 434 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 435 |
+
[36m(WorkerDict pid=1206072)[0m "rope_theta": 1000000.0,
|
| 436 |
+
[36m(WorkerDict pid=1206072)[0m "sliding_window": null,
|
| 437 |
+
[36m(WorkerDict pid=1206072)[0m "tie_word_embeddings": true,
|
| 438 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 439 |
+
[36m(WorkerDict pid=1206072)[0m "use_cache": true,
|
| 440 |
+
[36m(WorkerDict pid=1206072)[0m "use_sliding_window": false,
|
| 441 |
+
[36m(WorkerDict pid=1206072)[0m "video_token_id": null,
|
| 442 |
+
[36m(WorkerDict pid=1206072)[0m "vision_end_token_id": 151653,
|
| 443 |
+
[36m(WorkerDict pid=1206072)[0m "vision_start_token_id": 151652,
|
| 444 |
+
[36m(WorkerDict pid=1206072)[0m "vision_token_id": 151654,
|
| 445 |
+
[36m(WorkerDict pid=1206072)[0m "vocab_size": 151936
|
| 446 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 447 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 448 |
+
[36m(WorkerDict pid=1206072)[0m "transformers_version": "4.54.1",
|
| 449 |
+
[36m(WorkerDict pid=1206072)[0m "use_cache": true,
|
| 450 |
+
[36m(WorkerDict pid=1206072)[0m "use_sliding_window": false,
|
| 451 |
+
[36m(WorkerDict pid=1206072)[0m "video_token_id": 151656,
|
| 452 |
+
[36m(WorkerDict pid=1206072)[0m "vision_config": {
|
| 453 |
+
[36m(WorkerDict pid=1206072)[0m "depth": 32,
|
| 454 |
+
[36m(WorkerDict pid=1206072)[0m "fullatt_block_indexes": [
|
| 455 |
+
[36m(WorkerDict pid=1206072)[0m 7,
|
| 456 |
+
[36m(WorkerDict pid=1206072)[0m 15,
|
| 457 |
+
[36m(WorkerDict pid=1206072)[0m 23,
|
| 458 |
+
[36m(WorkerDict pid=1206072)[0m 31
|
| 459 |
+
[36m(WorkerDict pid=1206072)[0m ],
|
| 460 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_act": "silu",
|
| 461 |
+
[36m(WorkerDict pid=1206072)[0m "hidden_size": 1280,
|
| 462 |
+
[36m(WorkerDict pid=1206072)[0m "in_channels": 3,
|
| 463 |
+
[36m(WorkerDict pid=1206072)[0m "in_chans": 3,
|
| 464 |
+
[36m(WorkerDict pid=1206072)[0m "initializer_range": 0.02,
|
| 465 |
+
[36m(WorkerDict pid=1206072)[0m "intermediate_size": 3420,
|
| 466 |
+
[36m(WorkerDict pid=1206072)[0m "model_type": "qwen2_5_vl",
|
| 467 |
+
[36m(WorkerDict pid=1206072)[0m "num_heads": 16,
|
| 468 |
+
[36m(WorkerDict pid=1206072)[0m "out_hidden_size": 2048,
|
| 469 |
+
[36m(WorkerDict pid=1206072)[0m "patch_size": 14,
|
| 470 |
+
[36m(WorkerDict pid=1206072)[0m "spatial_merge_size": 2,
|
| 471 |
+
[36m(WorkerDict pid=1206072)[0m "spatial_patch_size": 14,
|
| 472 |
+
[36m(WorkerDict pid=1206072)[0m "temporal_patch_size": 2,
|
| 473 |
+
[36m(WorkerDict pid=1206072)[0m "tokens_per_second": 2,
|
| 474 |
+
[36m(WorkerDict pid=1206072)[0m "torch_dtype": "float32",
|
| 475 |
+
[36m(WorkerDict pid=1206072)[0m "window_size": 112
|
| 476 |
+
[36m(WorkerDict pid=1206072)[0m },
|
| 477 |
+
[36m(WorkerDict pid=1206072)[0m "vision_end_token_id": 151653,
|
| 478 |
+
[36m(WorkerDict pid=1206072)[0m "vision_start_token_id": 151652,
|
| 479 |
+
[36m(WorkerDict pid=1206072)[0m "vision_token_id": 151654,[36m(WorkerDict pid=1206072)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLForConditionalGeneration is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 480 |
+
[36m(WorkerDict pid=1206072)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 481 |
+
[36m(WorkerDict pid=1206072)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VisionTransformerPretrainedModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 482 |
+
[36m(WorkerDict pid=1206072)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 483 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 484 |
+
[36m(WorkerDict pid=1208116)[0m
|
| 485 |
+
[36m(WorkerDict pid=1208116)[0m [rank1]:[W615 11:26:41.958300854 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 1] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 486 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 487 |
+
[36m(WorkerDict pid=1208116)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`[32m [repeated 4x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/user-guides/configure-logging.html#log-deduplication for more options.)[0m
|
| 488 |
+
[36m(WorkerDict pid=1208116)[0m
|
| 489 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 490 |
+
[36m(WorkerDict pid=1208116)[0m
|
| 491 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 492 |
+
[36m(WorkerDict pid=1208116)[0m
|
| 493 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 494 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 495 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 496 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 497 |
+
[36m(WorkerDict pid=1208116)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 498 |
+
[36m(WorkerDict pid=1208116)[0m warnings.warn(
|
| 499 |
+
|
| 500 |
+
[36m(WorkerDict pid=1206072)[0m "vocab_size": 151936
|
| 501 |
+
[36m(WorkerDict pid=1206072)[0m }
|
| 502 |
+
[36m(WorkerDict pid=1206072)[0m
|
| 503 |
+
[36m(WorkerDict pid=1206072)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 504 |
+
[36m(WorkerDict pid=1206072)[0m wrap_policy: functools.partial(<function _or_policy at 0x7cebba2d45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7cebba2d44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 505 |
+
[36m(WorkerDict pid=1208116)[0m Actor use_remove_padding=False
|
| 506 |
+
[36m(WorkerDict pid=1206072)[0m Total steps: 472, num_warmup_steps: 0
|
| 507 |
+
[36m(WorkerDict pid=1206072)[0m Actor use_remove_padding=False
|
| 508 |
+
[36m(WorkerDict pid=1206072)[0m Before building vllm rollout, memory allocated (GB): 7.027174949645996, memory reserved (GB): 12.189453125
|
| 509 |
+
[36m(WorkerDict pid=1208116)[0m WARNING 06-15 11:29:28 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x799fe047bf50>
|
| 510 |
+
[36m(WorkerDict pid=1208116)[0m wrap_policy: functools.partial(<function _or_policy at 0x79a367cd45e0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x79a367cd44a0>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 511 |
+
[36m(WorkerDict pid=1208116)[0m Total steps: 472, num_warmup_steps: 0
|
| 512 |
+
[36m(WorkerDict pid=1208116)[0m Actor use_remove_padding=False
|
| 513 |
+
[36m(WorkerDict pid=1208116)[0m WARNING 06-15 11:29:36 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 514 |
+
[36m(WorkerDict pid=1206072)[0m WARNING 06-15 11:29:28 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x7ce8382d5cd0>
|
| 515 |
+
[36m(WorkerDict pid=1208116)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 516 |
+
[36m(WorkerDict pid=1206072)[0m WARNING 06-15 11:29:37 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 517 |
+
[36m(WorkerDict pid=1206072)[0m After building vllm rollout, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 518 |
+
[36m(WorkerDict pid=1206072)[0m After building sharding manager, memory allocated (GB): 13.633802890777588, memory reserved (GB): 15.04296875
|
| 519 |
+
[36m(TaskRunner pid=1203084)[0m Using LocalLogger is deprecated. The constructor API will change
|
| 520 |
+
[36m(TaskRunner pid=1203084)[0m validation generation end
|
| 521 |
+
[36m(TaskRunner pid=1203084)[0m [prompt] <|im_start|>system
|
| 522 |
+
[36m(TaskRunner pid=1203084)[0m You are a helpful assistant.<|im_end|>
|
| 523 |
+
[36m(TaskRunner pid=1203084)[0m <|im_start|>user
|
| 524 |
+
[36m(TaskRunner pid=1203084)[0m <|vision_start|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|image_pad|><|vision_end|>
|
| 525 |
+
[36m(TaskRunner pid=1203084)[0m You are an expert in univariate time-series anomaly detection. The figure shows a single-channel time series of length 64 from the "Server-IOPS" domain.
|
| 526 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 527 |
+
[36m(TaskRunner pid=1203084)[0m Decide whether the series is normal or contains an anomaly. If it is anomalous, choose the single most appropriate anomaly type from the following 15 categories (format `id: name — description`):
|
| 528 |
+
[36m(TaskRunner pid=1203084)[0m 0: Normal Sequence — There are no abnormal situations in this time series.
|
| 529 |
+
[36m(TaskRunner pid=1203084)[0m 1: Point Anomaly — A single data point significantly deviates from the local or global pattern of the sequence.
|
| 530 |
+
[36m(TaskRunner pid=1203084)[0m 2: Periodic Change Anomaly — The original periodic pattern is disrupted, e.g. the period is broken or the amplitude becomes anomalous.
|
| 531 |
+
[36m(TaskRunner pid=1203084)[0m 3: Trend Change Anomaly — A sudden change in the long-term trend of the time series.
|
| 532 |
+
[36m(TaskRunner pid=1203084)[0m 4: Change Point Anomaly — Statistical properties (e.g. mean, variance) change abruptly at certain points.
|
| 533 |
+
[36m(TaskRunner pid=1203084)[0m 5: Distributional Change Anomaly — The statistical distribution of the time series changes significantly.
|
| 534 |
+
[36m(TaskRunner pid=1203084)[0m 6: Amplitude Anomaly — The amplitude of data points exceeds the normal upper and lower bounds.
|
| 535 |
+
[36m(TaskRunner pid=1203084)[0m 7: Pattern Change Anomaly — The pattern of the time series suddenly changes from one form to another.
|
| 536 |
+
[36m(TaskRunner pid=1203084)[0m 8: Sparse Anomaly — Isolated anomalous patterns occasionally appear in a long time series.
|
| 537 |
+
[36m(TaskRunner pid=1203084)[0m 9: Repeated Value Anomaly — Continuous or intermittent repeated values disrupt the normal fluctuation pattern.
|
| 538 |
+
[36m(TaskRunner pid=1203084)[0m 10: Sudden Flatline Anomaly — The time series suddenly becomes a flat line with no normal fluctuations.
|
| 539 |
+
[36m(TaskRunner pid=1203084)[0m 11: Drift Anomaly — The data gradually drifts away from the normal level.
|
| 540 |
+
[36m(TaskRunner pid=1203084)[0m 12: Sudden Spike Anomaly — The data suddenly spikes or drops within a short time and then returns to normal.
|
| 541 |
+
[36m(TaskRunner pid=1203084)[0m 13: Continuous Segment Anomaly — A continuous segment of data points deviates from the normal pattern.
|
| 542 |
+
[36m(TaskRunner pid=1203084)[0m 14: Nonlinear Pattern Anomaly — Nonlinear changes appear in the sequence, breaking the original linear rule.
|
| 543 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 544 |
+
[36m(TaskRunner pid=1203084)[0m Reason step by step inside <think>...</think> based on the visual shape of the series, then output exactly one line with your final answer:
|
| 545 |
+
[36m(TaskRunner pid=1203084)[0m <class>one exact category name from the list above</class>
|
| 546 |
+
[36m(TaskRunner pid=1203084)[0m If the series is normal, use <class>Normal Sequence</class>.<|im_end|>
|
| 547 |
+
[36m(TaskRunner pid=1203084)[0m <|im_start|>assistant
|
| 548 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 549 |
+
[36m(TaskRunner pid=1203084)[0m [response] <think></think>
|
| 550 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 551 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 552 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 553 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 554 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 555 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 556 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 557 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 558 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 559 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 560 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 561 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 562 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 563 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 564 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 565 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 566 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 567 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 568 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 569 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 570 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 571 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 572 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 573 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 574 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 575 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 576 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 577 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 578 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 579 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 580 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 581 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 582 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 583 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 584 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 585 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 586 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 587 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 588 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 589 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 590 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 591 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 592 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 593 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 594 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 595 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 596 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 597 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 598 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 599 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 600 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 601 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 602 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 603 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 604 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 605 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 606 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 607 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 608 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 609 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 610 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 611 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 612 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 613 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 614 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 615 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 616 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 617 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 618 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 619 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 620 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 621 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 622 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 623 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 624 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 625 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 626 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 627 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 628 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 629 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 630 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 631 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 632 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 633 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 634 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 635 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 636 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 637 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 638 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 639 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 640 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 641 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 642 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 643 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 644 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 645 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 646 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 647 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 648 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 649 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 650 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 651 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 652 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 653 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 654 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 655 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 656 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 657 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 658 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 659 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 660 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 661 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 662 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 663 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 664 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 665 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 666 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 667 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 668 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 669 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 670 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 671 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 672 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 673 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 674 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 675 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 676 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 677 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 678 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 679 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 680 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 681 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 682 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 683 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 684 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 685 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 686 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 687 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 688 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 689 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 690 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 691 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 692 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 693 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 694 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 695 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 696 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 697 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 698 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 699 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 700 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 701 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 702 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 703 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 704 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 705 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 706 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 707 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 708 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 709 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 710 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 711 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 712 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 713 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 714 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 715 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 716 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 717 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 718 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 719 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 720 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 721 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 722 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 723 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 724 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 725 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 726 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 727 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 728 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 729 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 730 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 731 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 732 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 733 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 734 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 735 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 736 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 737 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 738 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 739 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 740 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 741 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 742 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 743 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 744 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 745 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 746 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 747 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 748 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 749 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 750 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 751 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 752 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 753 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 754 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 755 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 756 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 757 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 758 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 759 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 760 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 761 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 762 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 763 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 764 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 765 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 766 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 767 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 768 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 769 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 770 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 771 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 772 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 773 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 774 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 775 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 776 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 777 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 778 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 779 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 780 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 781 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 782 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 783 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 784 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 785 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 786 |
+
[36m(TaskRunner pid=1203084)[0m <class>Sudden Spike Anomaly</class>
|
| 787 |
+
[36m(TaskRunner pid=1203084)[0m
|
| 788 |
+
[36m(TaskRunner pid=1203084)[0m <think></think>
|
| 789 |
+
[36m(TaskRunner pid=1203084)[0m Detected new reward function format (3 return values).
|
| 790 |
+
[36m(WorkerDict pid=1206072)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 791 |
+
[36m(WorkerDict pid=1206072)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 792 |
+
[36m(WorkerDict pid=1206072)[0m warnings.warn(
|
| 793 |
+
[36m(TaskRunner pid=1203084)[0m ("Initial validation metrics: {'val/reward_score/timeseries_rats': "
|
| 794 |
+
[36m(TaskRunner pid=1203084)[0m "0.20000000298023224, 'val/metrics_overall/mean_affi precision': 0.0, "
|
| 795 |
+
[36m(TaskRunner pid=1203084)[0m "'val/metrics_overall/mean_affi recall': 0.0, 'val/metrics_overall/mean_affi "
|
| 796 |
+
[36m(TaskRunner pid=1203084)[0m "f1': 0.0, 'val/metrics_overall/mean_fmt_score': 1.0, "
|
| 797 |
+
[36m(TaskRunner pid=1203084)[0m "'val/metrics_overall/mean_class acc': 0.0, 'val/num_total_samples': 1, "
|
| 798 |
+
[36m(TaskRunner pid=1203084)[0m "'val/num_valid_samples': 1, 'val/num_invalid_predictions': 0, "
|
| 799 |
+
[36m(TaskRunner pid=1203084)[0m "'val/type_accuracy': 0.0, 'val/type_precision_macro': 0.0, "
|
| 800 |
+
[36m(TaskRunner pid=1203084)[0m "'val/type_recall_macro': 0.0, 'val/type_f1_macro': 0.0, "
|
| 801 |
+
[36m(TaskRunner pid=1203084)[0m "'val/binary_accuracy': 1.0, 'val/binary_precision_macro': 1.0, "
|
| 802 |
+
[36m(TaskRunner pid=1203084)[0m "'val/binary_recall_macro': 1.0, 'val/binary_f1_macro': 1.0}")
|
| 803 |
+
[36m(TaskRunner pid=1203084)[0m step:0 - val/reward_score/timeseries_rats:0.200 - val/metrics_overall/mean_affi precision:0.000 - val/metrics_overall/mean_affi recall:0.000 - val/metrics_overall/mean_affi f1:0.000 - val/metrics_overall/mean_fmt_score:1.000 - val/metrics_overall/mean_class acc:0.000 - val/num_total_samples:1.000 - val/num_valid_samples:1.000 - val/num_invalid_predictions:0.000 - val/type_accuracy:0.000 - val/type_precision_macro:0.000 - val/type_recall_macro:0.000 - val/type_f1_macro:0.000 - val/binary_accuracy:1.000 - val/binary_precision_macro:1.000 - val/binary_recall_macro:1.000 - val/binary_f1_macro:1.000
|
logs/rats_eval_chunked_check.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/rats_eval_full.log
ADDED
|
@@ -0,0 +1,569 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[log] saving full output to: /mnt/share01/sqk/AnomSeer/logs/rats_eval_full.log
|
| 2 |
+
[config] tuning=LoRA (r=16, alpha=16, target=q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj) | lr=1e-4 | gpus=2 | tensor-parallel (TP=2)
|
| 3 |
+
[config] model=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 4 |
+
[config] python=/home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python | stage=eval
|
| 5 |
+
[stage] EVAL (model=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct, gpus=2, tp=2)
|
| 6 |
+
2026-06-12 11:58:39,437 INFO worker.py:2012 -- Started a local Ray instance.
|
| 7 |
+
/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py:2051: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
|
| 8 |
+
warnings.warn(
|
| 9 |
+
[36m(TaskRunner pid=2835960)[0m {'actor_rollout_ref': {'actor': {'clip_ratio': 0.2,
|
| 10 |
+
[36m(TaskRunner pid=2835960)[0m 'entropy_coeff': 0.001,
|
| 11 |
+
[36m(TaskRunner pid=2835960)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 12 |
+
[36m(TaskRunner pid=2835960)[0m 'optimizer_offload': False,
|
| 13 |
+
[36m(TaskRunner pid=2835960)[0m 'param_offload': False,
|
| 14 |
+
[36m(TaskRunner pid=2835960)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 15 |
+
[36m(TaskRunner pid=2835960)[0m 'grad_clip': 1.0,
|
| 16 |
+
[36m(TaskRunner pid=2835960)[0m 'kl_loss_coef': 0.001,
|
| 17 |
+
[36m(TaskRunner pid=2835960)[0m 'kl_loss_type': 'low_var_kl',
|
| 18 |
+
[36m(TaskRunner pid=2835960)[0m 'optim': {'lr': 0.0001,
|
| 19 |
+
[36m(TaskRunner pid=2835960)[0m 'lr_warmup_steps': -1,
|
| 20 |
+
[36m(TaskRunner pid=2835960)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 21 |
+
[36m(TaskRunner pid=2835960)[0m 'min_lr_ratio': None,
|
| 22 |
+
[36m(TaskRunner pid=2835960)[0m 'total_training_steps': -1,
|
| 23 |
+
[36m(TaskRunner pid=2835960)[0m 'warmup_style': 'constant'},
|
| 24 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_epochs': 1,
|
| 25 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_max_token_len_per_gpu': 16384,
|
| 26 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_micro_batch_size': None,
|
| 27 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_micro_batch_size_per_gpu': 2,
|
| 28 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_mini_batch_size': 16,
|
| 29 |
+
[36m(TaskRunner pid=2835960)[0m 'shuffle': False,
|
| 30 |
+
[36m(TaskRunner pid=2835960)[0m 'strategy': 'fsdp',
|
| 31 |
+
[36m(TaskRunner pid=2835960)[0m 'ulysses_sequence_parallel_size': 1,
|
| 32 |
+
[36m(TaskRunner pid=2835960)[0m 'use_dynamic_bsz': False,
|
| 33 |
+
[36m(TaskRunner pid=2835960)[0m 'use_kl_loss': True,
|
| 34 |
+
[36m(TaskRunner pid=2835960)[0m 'use_torch_compile': True},
|
| 35 |
+
[36m(TaskRunner pid=2835960)[0m 'hybrid_engine': True,
|
| 36 |
+
[36m(TaskRunner pid=2835960)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 37 |
+
[36m(TaskRunner pid=2835960)[0m 'external_lib': None,
|
| 38 |
+
[36m(TaskRunner pid=2835960)[0m 'lora_alpha': 16,
|
| 39 |
+
[36m(TaskRunner pid=2835960)[0m 'lora_dropout': 0.0,
|
| 40 |
+
[36m(TaskRunner pid=2835960)[0m 'lora_rank': 16,
|
| 41 |
+
[36m(TaskRunner pid=2835960)[0m 'lora_target_modules': 'q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',
|
| 42 |
+
[36m(TaskRunner pid=2835960)[0m 'override_config': {},
|
| 43 |
+
[36m(TaskRunner pid=2835960)[0m 'path': '/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct',
|
| 44 |
+
[36m(TaskRunner pid=2835960)[0m 'use_remove_padding': False},
|
| 45 |
+
[36m(TaskRunner pid=2835960)[0m 'ref': {'fsdp_config': {'param_offload': True,
|
| 46 |
+
[36m(TaskRunner pid=2835960)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 47 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 48 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_micro_batch_size': None,
|
| 49 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 50 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_use_dynamic_bsz': False,
|
| 51 |
+
[36m(TaskRunner pid=2835960)[0m 'ulysses_sequence_parallel_size': 1},
|
| 52 |
+
[36m(TaskRunner pid=2835960)[0m 'rollout': {'disable_log_stats': True,
|
| 53 |
+
[36m(TaskRunner pid=2835960)[0m 'do_sample': True,
|
| 54 |
+
[36m(TaskRunner pid=2835960)[0m 'dtype': 'bfloat16',
|
| 55 |
+
[36m(TaskRunner pid=2835960)[0m 'enable_chunked_prefill': False,
|
| 56 |
+
[36m(TaskRunner pid=2835960)[0m 'enforce_eager': False,
|
| 57 |
+
[36m(TaskRunner pid=2835960)[0m 'free_cache_engine': False,
|
| 58 |
+
[36m(TaskRunner pid=2835960)[0m 'gpu_memory_utilization': 0.4,
|
| 59 |
+
[36m(TaskRunner pid=2835960)[0m 'ignore_eos': False,
|
| 60 |
+
[36m(TaskRunner pid=2835960)[0m 'load_format': 'dummy_dtensor',
|
| 61 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_max_token_len_per_gpu': 16384,
|
| 62 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_micro_batch_size': None,
|
| 63 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_micro_batch_size_per_gpu': 8,
|
| 64 |
+
[36m(TaskRunner pid=2835960)[0m 'log_prob_use_dynamic_bsz': False,
|
| 65 |
+
[36m(TaskRunner pid=2835960)[0m 'max_model_len': None,
|
| 66 |
+
[36m(TaskRunner pid=2835960)[0m 'max_num_batched_tokens': 8192,
|
| 67 |
+
[36m(TaskRunner pid=2835960)[0m 'max_num_seqs': 1024,
|
| 68 |
+
[36m(TaskRunner pid=2835960)[0m 'n': 5,
|
| 69 |
+
[36m(TaskRunner pid=2835960)[0m 'name': 'vllm',
|
| 70 |
+
[36m(TaskRunner pid=2835960)[0m 'prompt_length': 1024,
|
| 71 |
+
[36m(TaskRunner pid=2835960)[0m 'response_length': 1024,
|
| 72 |
+
[36m(TaskRunner pid=2835960)[0m 'temperature': 1.0,
|
| 73 |
+
[36m(TaskRunner pid=2835960)[0m 'tensor_model_parallel_size': 2,
|
| 74 |
+
[36m(TaskRunner pid=2835960)[0m 'top_k': -1,
|
| 75 |
+
[36m(TaskRunner pid=2835960)[0m 'top_p': 1,
|
| 76 |
+
[36m(TaskRunner pid=2835960)[0m 'use_fire_sampling': False,
|
| 77 |
+
[36m(TaskRunner pid=2835960)[0m 'val_kwargs': {'do_sample': False,
|
| 78 |
+
[36m(TaskRunner pid=2835960)[0m 'n': 1,
|
| 79 |
+
[36m(TaskRunner pid=2835960)[0m 'temperature': 0.4,
|
| 80 |
+
[36m(TaskRunner pid=2835960)[0m 'top_k': -1,
|
| 81 |
+
[36m(TaskRunner pid=2835960)[0m 'top_p': 1.0}}},
|
| 82 |
+
[36m(TaskRunner pid=2835960)[0m 'algorithm': {'adv_estimator': 'grpo',
|
| 83 |
+
[36m(TaskRunner pid=2835960)[0m 'gamma': 1.0,
|
| 84 |
+
[36m(TaskRunner pid=2835960)[0m 'kl_ctrl': {'kl_coef': 0.001, 'type': 'fixed'},
|
| 85 |
+
[36m(TaskRunner pid=2835960)[0m 'kl_penalty': 'kl',
|
| 86 |
+
[36m(TaskRunner pid=2835960)[0m 'lam': 1.0},
|
| 87 |
+
[36m(TaskRunner pid=2835960)[0m 'critic': {'cliprange_value': 0.5,
|
| 88 |
+
[36m(TaskRunner pid=2835960)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 89 |
+
[36m(TaskRunner pid=2835960)[0m 'forward_micro_batch_size': None,
|
| 90 |
+
[36m(TaskRunner pid=2835960)[0m 'forward_micro_batch_size_per_gpu': None,
|
| 91 |
+
[36m(TaskRunner pid=2835960)[0m 'grad_clip': 1.0,
|
| 92 |
+
[36m(TaskRunner pid=2835960)[0m 'model': {'enable_gradient_checkpointing': True,
|
| 93 |
+
[36m(TaskRunner pid=2835960)[0m 'external_lib': None,
|
| 94 |
+
[36m(TaskRunner pid=2835960)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 95 |
+
[36m(TaskRunner pid=2835960)[0m 'optimizer_offload': False,[36m(TaskRunner pid=2835960)[0m You have video processor config saved in `preprocessor.json` file which is deprecated. Video processor configs should be saved in their own `video_preprocessor.json` file. You can rename the file or load and save the processor back which renames it automatically. Loading from `preprocessor.json` will be removed in v5.0.
|
| 96 |
+
[36m(TaskRunner pid=2835960)[0m DeprecationWarning: `ray.state.available_resources_per_node` is a private attribute and access will be removed in a future Ray version.
|
| 97 |
+
[36m(WorkerDict pid=2839719)[0m The image processor of type `Qwen2VLImageProcessor` is now loaded as a fast processor by default, even if the model checkpoint was saved with a slow processor. This is a breaking change and may produce slightly different outputs. To continue using the slow processor, instantiate this class with `use_fast=False`. Note that this behavior will be extended to all models in a future release.
|
| 98 |
+
[36m(WorkerDict pid=2839719)[0m You have video processor config saved in `preprocessor.json` file which is deprecated. Video processor configs should be saved in their own `video_preprocessor.json` file. You can rename the file or load and save the processor back which renames it automatically. Loading from `preprocessor.json` will be removed in v5.0.
|
| 99 |
+
|
| 100 |
+
[36m(TaskRunner pid=2835960)[0m 'param_offload': False,
|
| 101 |
+
[36m(TaskRunner pid=2835960)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 102 |
+
[36m(TaskRunner pid=2835960)[0m 'override_config': {},
|
| 103 |
+
[36m(TaskRunner pid=2835960)[0m 'path': '~/models/deepseek-llm-7b-chat',
|
| 104 |
+
[36m(TaskRunner pid=2835960)[0m 'tokenizer_path': '/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct',
|
| 105 |
+
[36m(TaskRunner pid=2835960)[0m 'use_remove_padding': False},
|
| 106 |
+
[36m(TaskRunner pid=2835960)[0m 'optim': {'lr': 1e-05,
|
| 107 |
+
[36m(TaskRunner pid=2835960)[0m 'lr_warmup_steps_ratio': 0.0,
|
| 108 |
+
[36m(TaskRunner pid=2835960)[0m 'min_lr_ratio': None,
|
| 109 |
+
[36m(TaskRunner pid=2835960)[0m 'total_training_steps': -1,
|
| 110 |
+
[36m(TaskRunner pid=2835960)[0m 'warmup_style': 'constant'},
|
| 111 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_epochs': 1,
|
| 112 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_max_token_len_per_gpu': 32768,
|
| 113 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_micro_batch_size': None,
|
| 114 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_micro_batch_size_per_gpu': None,
|
| 115 |
+
[36m(TaskRunner pid=2835960)[0m 'ppo_mini_batch_size': 16,
|
| 116 |
+
[36m(TaskRunner pid=2835960)[0m 'shuffle': False,
|
| 117 |
+
[36m(TaskRunner pid=2835960)[0m 'strategy': 'fsdp',
|
| 118 |
+
[36m(TaskRunner pid=2835960)[0m 'ulysses_sequence_parallel_size': 1,
|
| 119 |
+
[36m(TaskRunner pid=2835960)[0m 'use_dynamic_bsz': False},
|
| 120 |
+
[36m(TaskRunner pid=2835960)[0m 'custom_reward_function': {'name': 'compute_score', 'path': None},
|
| 121 |
+
[36m(TaskRunner pid=2835960)[0m 'data': {'filter_overlong_prompts': True,
|
| 122 |
+
[36m(TaskRunner pid=2835960)[0m 'image_key': 'images',
|
| 123 |
+
[36m(TaskRunner pid=2835960)[0m 'max_prompt_length': 1024,
|
| 124 |
+
[36m(TaskRunner pid=2835960)[0m 'max_response_length': 1024,
|
| 125 |
+
[36m(TaskRunner pid=2835960)[0m 'prompt_key': 'prompt',
|
| 126 |
+
[36m(TaskRunner pid=2835960)[0m 'return_raw_chat': False,
|
| 127 |
+
[36m(TaskRunner pid=2835960)[0m 'return_raw_input_ids': False,
|
| 128 |
+
[36m(TaskRunner pid=2835960)[0m 'shuffle': True,
|
| 129 |
+
[36m(TaskRunner pid=2835960)[0m 'tokenizer': None,
|
| 130 |
+
[36m(TaskRunner pid=2835960)[0m 'train_batch_size': 16,
|
| 131 |
+
[36m(TaskRunner pid=2835960)[0m 'train_files': './data/rats_uni_processed/train_quarter.parquet',
|
| 132 |
+
[36m(TaskRunner pid=2835960)[0m 'truncation': 'error',
|
| 133 |
+
[36m(TaskRunner pid=2835960)[0m 'val_batch_size': None,
|
| 134 |
+
[36m(TaskRunner pid=2835960)[0m 'val_files': './data/rats_uni_processed/test_full.parquet'},
|
| 135 |
+
[36m(TaskRunner pid=2835960)[0m 'reward_model': {'enable': False,
|
| 136 |
+
[36m(TaskRunner pid=2835960)[0m 'forward_max_token_len_per_gpu': 32768,
|
| 137 |
+
[36m(TaskRunner pid=2835960)[0m 'max_length': None,
|
| 138 |
+
[36m(TaskRunner pid=2835960)[0m 'micro_batch_size': None,
|
| 139 |
+
[36m(TaskRunner pid=2835960)[0m 'micro_batch_size_per_gpu': None,
|
| 140 |
+
[36m(TaskRunner pid=2835960)[0m 'model': {'external_lib': None,
|
| 141 |
+
[36m(TaskRunner pid=2835960)[0m 'fsdp_config': {'fsdp_size': -1,
|
| 142 |
+
[36m(TaskRunner pid=2835960)[0m 'param_offload': False,
|
| 143 |
+
[36m(TaskRunner pid=2835960)[0m 'wrap_policy': {'min_num_params': 0}},
|
| 144 |
+
[36m(TaskRunner pid=2835960)[0m 'input_tokenizer': '/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct',
|
| 145 |
+
[36m(TaskRunner pid=2835960)[0m 'path': '~/models/FsfairX-LLaMA3-RM-v0.1',
|
| 146 |
+
[36m(TaskRunner pid=2835960)[0m 'use_remove_padding': False},
|
| 147 |
+
[36m(TaskRunner pid=2835960)[0m 'reward_manager': 'naive',
|
| 148 |
+
[36m(TaskRunner pid=2835960)[0m 'strategy': 'fsdp',
|
| 149 |
+
[36m(TaskRunner pid=2835960)[0m 'ulysses_sequence_parallel_size': 1,
|
| 150 |
+
[36m(TaskRunner pid=2835960)[0m 'use_dynamic_bsz': False},
|
| 151 |
+
[36m(TaskRunner pid=2835960)[0m 'trainer': {'balance_batch': True,
|
| 152 |
+
[36m(TaskRunner pid=2835960)[0m 'critic_warmup': 0,
|
| 153 |
+
[36m(TaskRunner pid=2835960)[0m 'default_hdfs_dir': None,
|
| 154 |
+
[36m(TaskRunner pid=2835960)[0m 'default_local_dir': 'checkpoints/anomseer/anomseer_rats_uni_2gpu',
|
| 155 |
+
[36m(TaskRunner pid=2835960)[0m 'del_local_ckpt_after_load': False,
|
| 156 |
+
[36m(TaskRunner pid=2835960)[0m 'experiment_name': 'anomseer_rats_uni_2gpu',
|
| 157 |
+
[36m(TaskRunner pid=2835960)[0m 'logger': ['console'],
|
| 158 |
+
[36m(TaskRunner pid=2835960)[0m 'n_gpus_per_node': 2,
|
| 159 |
+
[36m(TaskRunner pid=2835960)[0m 'nnodes': 1,
|
| 160 |
+
[36m(TaskRunner pid=2835960)[0m 'project_name': 'anomseer',
|
| 161 |
+
[36m(TaskRunner pid=2835960)[0m 'remove_previous_ckpt_in_save': False,
|
| 162 |
+
[36m(TaskRunner pid=2835960)[0m 'resume_from_path': False,
|
| 163 |
+
[36m(TaskRunner pid=2835960)[0m 'resume_mode': 'auto',
|
| 164 |
+
[36m(TaskRunner pid=2835960)[0m 'save_freq': 500,
|
| 165 |
+
[36m(TaskRunner pid=2835960)[0m 'test_freq': 10,
|
| 166 |
+
[36m(TaskRunner pid=2835960)[0m 'total_epochs': 1,
|
| 167 |
+
[36m(TaskRunner pid=2835960)[0m 'total_training_steps': None,
|
| 168 |
+
[36m(TaskRunner pid=2835960)[0m 'val_before_train': True,
|
| 169 |
+
[36m(TaskRunner pid=2835960)[0m 'val_generations_to_log_to_wandb': 0,
|
| 170 |
+
[36m(TaskRunner pid=2835960)[0m 'val_only': True},
|
| 171 |
+
[36m(TaskRunner pid=2835960)[0m 'ts': {'adv_mix': 0.3,
|
| 172 |
+
[36m(TaskRunner pid=2835960)[0m 'clip_pooling': 'mean',
|
| 173 |
+
[36m(TaskRunner pid=2835960)[0m 'clip_temperature': 1.0,
|
| 174 |
+
[36m(TaskRunner pid=2835960)[0m 'ot_eps': 0.08,
|
| 175 |
+
[36m(TaskRunner pid=2835960)[0m 'ot_n_iter': 50,
|
| 176 |
+
[36m(TaskRunner pid=2835960)[0m 'similarity_method': 'ot',
|
| 177 |
+
[36m(TaskRunner pid=2835960)[0m 'use_sem_orth': True}}
|
| 178 |
+
[36m(TaskRunner pid=2835960)[0m reward_manager_name: anomts
|
| 179 |
+
[36m(TaskRunner pid=2835960)[0m [validate_config] All configuration checks passed successfully!
|
| 180 |
+
[36m(TaskRunner pid=2835960)[0m dataset len: 7566
|
| 181 |
+
[36m(TaskRunner pid=2835960)[0m filter dataset len: 7566
|
| 182 |
+
[36m(TaskRunner pid=2835960)[0m dataset len: 6034
|
| 183 |
+
[36m(TaskRunner pid=2835960)[0m filter dataset len: 6034
|
| 184 |
+
[36m(TaskRunner pid=2835960)[0m Size of train dataloader: 472
|
| 185 |
+
[36m(TaskRunner pid=2835960)[0m Total training steps: 472
|
| 186 |
+
[36m(WorkerDict pid=2839719)[0m Model config after override: Qwen2_5_VLConfig {
|
| 187 |
+
[36m(WorkerDict pid=2839719)[0m "architectures": [
|
| 188 |
+
[36m(WorkerDict pid=2839719)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 189 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 190 |
+
[36m(WorkerDict pid=2839719)[0m "attention_dropout": 0.0,
|
| 191 |
+
[36m(WorkerDict pid=2839719)[0m "eos_token_id": 151645,
|
| 192 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",
|
| 193 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 2048,
|
| 194 |
+
[36m(WorkerDict pid=2839719)[0m "image_token_id": 151655,
|
| 195 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 196 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 11008,
|
| 197 |
+
[36m(WorkerDict pid=2839719)[0m "max_position_embeddings": 128000,
|
| 198 |
+
[36m(WorkerDict pid=2839719)[0m "max_window_layers": 70,
|
| 199 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl",
|
| 200 |
+
[36m(WorkerDict pid=2839719)[0m "num_attention_heads": 16,
|
| 201 |
+
[36m(WorkerDict pid=2839719)[0m "num_hidden_layers": 36,
|
| 202 |
+
[36m(WorkerDict pid=2839719)[0m "num_key_value_heads": 2,
|
| 203 |
+
[36m(WorkerDict pid=2839719)[0m "pad_token_id": 151643,
|
| 204 |
+
[36m(WorkerDict pid=2839719)[0m "rms_norm_eps": 1e-06,
|
| 205 |
+
[36m(WorkerDict pid=2839719)[0m "rope_scaling": {
|
| 206 |
+
[36m(WorkerDict pid=2839719)[0m "mrope_section": [
|
| 207 |
+
[36m(WorkerDict pid=2839719)[0m 16,
|
| 208 |
+
[36m(WorkerDict pid=2839719)[0m 24,
|
| 209 |
+
[36m(WorkerDict pid=2839719)[0m 24
|
| 210 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 211 |
+
[36m(WorkerDict pid=2839719)[0m "rope_type": "default",[36m(WorkerDict pid=2839719)[0m
|
| 212 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 213 |
+
[36m(WorkerDict pid=2841587)[0m The image processor of type `Qwen2VLImageProcessor` is now loaded as a fast processor by default, even if the model checkpoint was saved with a slow processor. This is a breaking change and may produce slightly different outputs. To continue using the slow processor, instantiate this class with `use_fast=False`. Note that this behavior will be extended to all models in a future release.
|
| 214 |
+
[36m(WorkerDict pid=2841587)[0m You have video processor config saved in `preprocessor.json` file which is deprecated. Video processor configs should be saved in their own `video_preprocessor.json` file. You can rename the file or load and save the processor back which renames it automatically. Loading from `preprocessor.json` will be removed in v5.0.
|
| 215 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 216 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 217 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 218 |
+
[36m(WorkerDict pid=2839719)[0m [rank0]:[W612 11:59:36.925129932 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 0] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 219 |
+
[36m(WorkerDict pid=2841587)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLForConditionalGeneration is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 220 |
+
[36m(WorkerDict pid=2841587)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 221 |
+
[36m(WorkerDict pid=2841587)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VisionTransformerPretrainedModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`
|
| 222 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 223 |
+
[36m(WorkerDict pid=2841587)[0m [rank1]:[W612 11:59:36.938180582 ProcessGroupNCCL.cpp:4561] [PG ID 0 PG GUID 0 Rank 1] using GPU 0 to perform barrier as devices used by this process are currently unknown. This can potentially cause a hang if this rank to GPU mapping is incorrect. Specify device_ids in barrier() to force use of a particular device, or call init_process_group() with a device_id.
|
| 224 |
+
|
| 225 |
+
[36m(WorkerDict pid=2839719)[0m "type": "default"
|
| 226 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 227 |
+
[36m(WorkerDict pid=2839719)[0m "rope_theta": 1000000.0,
|
| 228 |
+
[36m(WorkerDict pid=2839719)[0m "sliding_window": 32768,
|
| 229 |
+
[36m(WorkerDict pid=2839719)[0m "text_config": {
|
| 230 |
+
[36m(WorkerDict pid=2839719)[0m "architectures": [
|
| 231 |
+
[36m(WorkerDict pid=2839719)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 232 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 233 |
+
[36m(WorkerDict pid=2839719)[0m "attention_dropout": 0.0,
|
| 234 |
+
[36m(WorkerDict pid=2839719)[0m "bos_token_id": 151643,
|
| 235 |
+
[36m(WorkerDict pid=2839719)[0m "eos_token_id": 151645,
|
| 236 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",
|
| 237 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 2048,
|
| 238 |
+
[36m(WorkerDict pid=2839719)[0m "image_token_id": null,
|
| 239 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 240 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 11008,
|
| 241 |
+
[36m(WorkerDict pid=2839719)[0m "layer_types": [
|
| 242 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 243 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 244 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 245 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 246 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 247 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 248 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 249 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 250 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 251 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 252 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 253 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 254 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 255 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 256 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 257 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 258 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 259 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 260 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 261 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 262 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 263 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 264 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 265 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 266 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 267 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 268 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 269 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 270 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 271 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 272 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 273 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 274 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 275 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 276 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 277 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention"
|
| 278 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 279 |
+
[36m(WorkerDict pid=2839719)[0m "max_position_embeddings": 128000,
|
| 280 |
+
[36m(WorkerDict pid=2839719)[0m "max_window_layers": 70,
|
| 281 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl_text",
|
| 282 |
+
[36m(WorkerDict pid=2839719)[0m "num_attention_heads": 16,
|
| 283 |
+
[36m(WorkerDict pid=2839719)[0m "num_hidden_layers": 36,
|
| 284 |
+
[36m(WorkerDict pid=2839719)[0m "num_key_value_heads": 2,
|
| 285 |
+
[36m(WorkerDict pid=2839719)[0m "rms_norm_eps": 1e-06,
|
| 286 |
+
[36m(WorkerDict pid=2839719)[0m "rope_scaling": {
|
| 287 |
+
[36m(WorkerDict pid=2839719)[0m "mrope_section": [
|
| 288 |
+
[36m(WorkerDict pid=2839719)[0m 16,
|
| 289 |
+
[36m(WorkerDict pid=2839719)[0m 24,
|
| 290 |
+
[36m(WorkerDict pid=2839719)[0m 24
|
| 291 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 292 |
+
[36m(WorkerDict pid=2839719)[0m "rope_type": "default",
|
| 293 |
+
[36m(WorkerDict pid=2839719)[0m "type": "default"
|
| 294 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 295 |
+
[36m(WorkerDict pid=2839719)[0m "rope_theta": 1000000.0,
|
| 296 |
+
[36m(WorkerDict pid=2839719)[0m "sliding_window": null,
|
| 297 |
+
[36m(WorkerDict pid=2839719)[0m "tie_word_embeddings": true,
|
| 298 |
+
[36m(WorkerDict pid=2839719)[0m "torch_dtype": "bfloat16",
|
| 299 |
+
[36m(WorkerDict pid=2839719)[0m "use_cache": true,
|
| 300 |
+
[36m(WorkerDict pid=2839719)[0m "use_sliding_window": false,
|
| 301 |
+
[36m(WorkerDict pid=2839719)[0m "video_token_id": null,
|
| 302 |
+
[36m(WorkerDict pid=2839719)[0m "vision_end_token_id": 151653,
|
| 303 |
+
[36m(WorkerDict pid=2839719)[0m "vision_start_token_id": 151652,
|
| 304 |
+
[36m(WorkerDict pid=2839719)[0m "vision_token_id": 151654,
|
| 305 |
+
[36m(WorkerDict pid=2839719)[0m "vocab_size": 151936
|
| 306 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 307 |
+
[36m(WorkerDict pid=2839719)[0m "torch_dtype": "bfloat16",
|
| 308 |
+
[36m(WorkerDict pid=2839719)[0m "transformers_version": "4.54.1",
|
| 309 |
+
[36m(WorkerDict pid=2839719)[0m "use_cache": true,
|
| 310 |
+
[36m(WorkerDict pid=2839719)[0m "use_sliding_window": false,
|
| 311 |
+
[36m(WorkerDict pid=2839719)[0m "video_token_id": 151656,
|
| 312 |
+
[36m(WorkerDict pid=2839719)[0m "vision_config": {
|
| 313 |
+
[36m(WorkerDict pid=2839719)[0m "depth": 32,
|
| 314 |
+
[36m(WorkerDict pid=2839719)[0m "fullatt_block_indexes": [
|
| 315 |
+
[36m(WorkerDict pid=2839719)[0m 7,
|
| 316 |
+
[36m(WorkerDict pid=2839719)[0m 15,
|
| 317 |
+
[36m(WorkerDict pid=2839719)[0m 23,
|
| 318 |
+
[36m(WorkerDict pid=2839719)[0m 31
|
| 319 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 320 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",
|
| 321 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 1280,
|
| 322 |
+
[36m(WorkerDict pid=2839719)[0m "in_channels": 3,
|
| 323 |
+
[36m(WorkerDict pid=2839719)[0m "in_chans": 3,
|
| 324 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 325 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 3420,
|
| 326 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl",
|
| 327 |
+
[36m(WorkerDict pid=2839719)[0m "num_heads": 16,
|
| 328 |
+
[36m(WorkerDict pid=2839719)[0m "out_hidden_size": 2048,
|
| 329 |
+
[36m(WorkerDict pid=2839719)[0m "patch_size": 14,
|
| 330 |
+
[36m(WorkerDict pid=2839719)[0m "spatial_merge_size": 2,
|
| 331 |
+
[36m(WorkerDict pid=2839719)[0m "spatial_patch_size": 14,
|
| 332 |
+
[36m(WorkerDict pid=2839719)[0m "temporal_patch_size": 2,
|
| 333 |
+
[36m(WorkerDict pid=2839719)[0m "tokens_per_second": 2,
|
| 334 |
+
[36m(WorkerDict pid=2839719)[0m "window_size": 112
|
| 335 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 336 |
+
[36m(WorkerDict pid=2839719)[0m "vision_end_token_id": 151653,
|
| 337 |
+
[36m(WorkerDict pid=2839719)[0m "vision_start_token_id": 151652,
|
| 338 |
+
[36m(WorkerDict pid=2839719)[0m "vision_token_id": 151654,
|
| 339 |
+
[36m(WorkerDict pid=2839719)[0m "vocab_size": 151936
|
| 340 |
+
[36m(WorkerDict pid=2839719)[0m }
|
| 341 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 342 |
+
[36m(WorkerDict pid=2839719)[0m NCCL version 2.21.5+cuda12.4
|
| 343 |
+
[36m(WorkerDict pid=2839719)[0m Qwen2_5_VLForConditionalGeneration contains 3.75B parameters
|
| 344 |
+
[36m(WorkerDict pid=2839719)[0m wrap_policy: functools.partial(<function _or_policy at 0x7e64dc2c84a0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7e64dc2c8360>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 345 |
+
[36m(WorkerDict pid=2839719)[0m Actor use_remove_padding=False
|
| 346 |
+
[36m(WorkerDict pid=2841587)[0m wrap_policy: functools.partial(<function _or_policy at 0x7bba8c2c44a0>, policies=[functools.partial(<function transformer_auto_wrap_policy at 0x7bba8c2c4360>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 347 |
+
[36m(WorkerDict pid=2839719)[0m Model config after override: Qwen2_5_VLConfig {
|
| 348 |
+
[36m(WorkerDict pid=2839719)[0m "architectures": [
|
| 349 |
+
[36m(WorkerDict pid=2839719)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 350 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 351 |
+
[36m(WorkerDict pid=2839719)[0m "attention_dropout": 0.0,
|
| 352 |
+
[36m(WorkerDict pid=2839719)[0m "eos_token_id": 151645,
|
| 353 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",[36m(WorkerDict pid=2839719)[0m
|
| 354 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 355 |
+
[36m(WorkerDict pid=2841587)[0m Flash Attention 2 only supports torch.float16 and torch.bfloat16 dtypes, but the current dype in Qwen2_5_VLTextModel is torch.float32. You should run training or inference using Automatic Mixed-Precision via the `with torch.autocast(device_type='torch_device'):` decorator, or load the model with the `torch_dtype` argument. Example: `model = AutoModel.from_pretrained("openai/whisper-tiny", attn_implementation="flash_attention_2", torch_dtype=torch.float16)`[32m [repeated 5x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/user-guides/configure-logging.html#log-deduplication for more options.)[0m
|
| 356 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 357 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 358 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 359 |
+
|
| 360 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 2048,
|
| 361 |
+
[36m(WorkerDict pid=2839719)[0m "image_token_id": 151655,
|
| 362 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 363 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 11008,
|
| 364 |
+
[36m(WorkerDict pid=2839719)[0m "max_position_embeddings": 128000,
|
| 365 |
+
[36m(WorkerDict pid=2839719)[0m "max_window_layers": 70,
|
| 366 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl",
|
| 367 |
+
[36m(WorkerDict pid=2839719)[0m "num_attention_heads": 16,
|
| 368 |
+
[36m(WorkerDict pid=2839719)[0m "num_hidden_layers": 36,
|
| 369 |
+
[36m(WorkerDict pid=2839719)[0m "num_key_value_heads": 2,
|
| 370 |
+
[36m(WorkerDict pid=2839719)[0m "pad_token_id": 151643,
|
| 371 |
+
[36m(WorkerDict pid=2839719)[0m "rms_norm_eps": 1e-06,
|
| 372 |
+
[36m(WorkerDict pid=2839719)[0m "rope_scaling": {
|
| 373 |
+
[36m(WorkerDict pid=2839719)[0m "mrope_section": [
|
| 374 |
+
[36m(WorkerDict pid=2839719)[0m 16,
|
| 375 |
+
[36m(WorkerDict pid=2839719)[0m 24,
|
| 376 |
+
[36m(WorkerDict pid=2839719)[0m 24
|
| 377 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 378 |
+
[36m(WorkerDict pid=2839719)[0m "rope_type": "default",
|
| 379 |
+
[36m(WorkerDict pid=2839719)[0m "type": "default"
|
| 380 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 381 |
+
[36m(WorkerDict pid=2839719)[0m "rope_theta": 1000000.0,
|
| 382 |
+
[36m(WorkerDict pid=2839719)[0m "sliding_window": 32768,
|
| 383 |
+
[36m(WorkerDict pid=2839719)[0m "text_config": {
|
| 384 |
+
[36m(WorkerDict pid=2839719)[0m "architectures": [
|
| 385 |
+
[36m(WorkerDict pid=2839719)[0m "Qwen2_5_VLForConditionalGeneration"
|
| 386 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 387 |
+
[36m(WorkerDict pid=2839719)[0m "attention_dropout": 0.0,
|
| 388 |
+
[36m(WorkerDict pid=2839719)[0m "bos_token_id": 151643,
|
| 389 |
+
[36m(WorkerDict pid=2839719)[0m "eos_token_id": 151645,
|
| 390 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",
|
| 391 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 2048,
|
| 392 |
+
[36m(WorkerDict pid=2839719)[0m "image_token_id": null,
|
| 393 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 394 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 11008,
|
| 395 |
+
[36m(WorkerDict pid=2839719)[0m "layer_types": [
|
| 396 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 397 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 398 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 399 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 400 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 401 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 402 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 403 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 404 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 405 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 406 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 407 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 408 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 409 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 410 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 411 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 412 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 413 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 414 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 415 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 416 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 417 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 418 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 419 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 420 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 421 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 422 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 423 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 424 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 425 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 426 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 427 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 428 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 429 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 430 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention",
|
| 431 |
+
[36m(WorkerDict pid=2839719)[0m "full_attention"
|
| 432 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 433 |
+
[36m(WorkerDict pid=2839719)[0m "max_position_embeddings": 128000,
|
| 434 |
+
[36m(WorkerDict pid=2839719)[0m "max_window_layers": 70,
|
| 435 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl_text",
|
| 436 |
+
[36m(WorkerDict pid=2839719)[0m "num_attention_heads": 16,
|
| 437 |
+
[36m(WorkerDict pid=2839719)[0m "num_hidden_layers": 36,
|
| 438 |
+
[36m(WorkerDict pid=2839719)[0m "num_key_value_heads": 2,
|
| 439 |
+
[36m(WorkerDict pid=2839719)[0m "rms_norm_eps": 1e-06,
|
| 440 |
+
[36m(WorkerDict pid=2839719)[0m "rope_scaling": {
|
| 441 |
+
[36m(WorkerDict pid=2839719)[0m "mrope_section": [
|
| 442 |
+
[36m(WorkerDict pid=2839719)[0m 16,
|
| 443 |
+
[36m(WorkerDict pid=2839719)[0m 24,
|
| 444 |
+
[36m(WorkerDict pid=2839719)[0m 24
|
| 445 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 446 |
+
[36m(WorkerDict pid=2839719)[0m "rope_type": "default",
|
| 447 |
+
[36m(WorkerDict pid=2839719)[0m "type": "default"
|
| 448 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 449 |
+
[36m(WorkerDict pid=2839719)[0m "rope_theta": 1000000.0,
|
| 450 |
+
[36m(WorkerDict pid=2839719)[0m "sliding_window": null,
|
| 451 |
+
[36m(WorkerDict pid=2839719)[0m "tie_word_embeddings": true,
|
| 452 |
+
[36m(WorkerDict pid=2839719)[0m "torch_dtype": "bfloat16",
|
| 453 |
+
[36m(WorkerDict pid=2839719)[0m "use_cache": true,
|
| 454 |
+
[36m(WorkerDict pid=2839719)[0m "use_sliding_window": false,
|
| 455 |
+
[36m(WorkerDict pid=2839719)[0m "video_token_id": null,
|
| 456 |
+
[36m(WorkerDict pid=2839719)[0m "vision_end_token_id": 151653,
|
| 457 |
+
[36m(WorkerDict pid=2839719)[0m "vision_start_token_id": 151652,
|
| 458 |
+
[36m(WorkerDict pid=2839719)[0m "vision_token_id": 151654,
|
| 459 |
+
[36m(WorkerDict pid=2839719)[0m "vocab_size": 151936
|
| 460 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 461 |
+
[36m(WorkerDict pid=2839719)[0m "torch_dtype": "bfloat16",
|
| 462 |
+
[36m(WorkerDict pid=2839719)[0m "transformers_version": "4.54.1",
|
| 463 |
+
[36m(WorkerDict pid=2839719)[0m "use_cache": true,
|
| 464 |
+
[36m(WorkerDict pid=2839719)[0m "use_sliding_window": false,
|
| 465 |
+
[36m(WorkerDict pid=2839719)[0m "video_token_id": 151656,
|
| 466 |
+
[36m(WorkerDict pid=2839719)[0m "vision_config": {
|
| 467 |
+
[36m(WorkerDict pid=2839719)[0m "depth": 32,
|
| 468 |
+
[36m(WorkerDict pid=2839719)[0m "fullatt_block_indexes": [
|
| 469 |
+
[36m(WorkerDict pid=2839719)[0m 7,
|
| 470 |
+
[36m(WorkerDict pid=2839719)[0m 15,
|
| 471 |
+
[36m(WorkerDict pid=2839719)[0m 23,
|
| 472 |
+
[36m(WorkerDict pid=2839719)[0m 31
|
| 473 |
+
[36m(WorkerDict pid=2839719)[0m ],
|
| 474 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_act": "silu",
|
| 475 |
+
[36m(WorkerDict pid=2839719)[0m "hidden_size": 1280,
|
| 476 |
+
[36m(WorkerDict pid=2839719)[0m "in_channels": 3,
|
| 477 |
+
[36m(WorkerDict pid=2839719)[0m "in_chans": 3,
|
| 478 |
+
[36m(WorkerDict pid=2839719)[0m "initializer_range": 0.02,
|
| 479 |
+
[36m(WorkerDict pid=2839719)[0m "intermediate_size": 3420,
|
| 480 |
+
[36m(WorkerDict pid=2839719)[0m "model_type": "qwen2_5_vl",
|
| 481 |
+
[36m(WorkerDict pid=2839719)[0m "num_heads": 16,
|
| 482 |
+
[36m(WorkerDict pid=2839719)[0m "out_hidden_size": 2048,
|
| 483 |
+
[36m(WorkerDict pid=2839719)[0m "patch_size": 14,
|
| 484 |
+
[36m(WorkerDict pid=2839719)[0m "spatial_merge_size": 2,
|
| 485 |
+
[36m(WorkerDict pid=2839719)[0m "spatial_patch_size": 14,
|
| 486 |
+
[36m(WorkerDict pid=2839719)[0m "temporal_patch_size": 2,
|
| 487 |
+
[36m(WorkerDict pid=2839719)[0m "tokens_per_second": 2,
|
| 488 |
+
[36m(WorkerDict pid=2839719)[0m "window_size": 112
|
| 489 |
+
[36m(WorkerDict pid=2839719)[0m },
|
| 490 |
+
[36m(WorkerDict pid=2839719)[0m "vision_end_token_id": 151653,
|
| 491 |
+
[36m(WorkerDict pid=2839719)[0m "vision_start_token_id": 151652,
|
| 492 |
+
[36m(WorkerDict pid=2839719)[0m "vision_token_id": 151654,
|
| 493 |
+
[36m(WorkerDict pid=2839719)[0m "vocab_size": 151936
|
| 494 |
+
[36m(WorkerDict pid=2839719)[0m }
|
| 495 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 496 |
+
[36m(WorkerDict pid=2839719)[0m trainable params: 37,152,768 || all params: 3,791,775,744 || trainable%: 0.9798
|
| 497 |
+
[36m(WorkerDict pid=2839719)[0m PeftModelForCausalLM contains 3.79B parameters
|
| 498 |
+
[36m(WorkerDict pid=2839719)[0m wrap_policy: functools.partial(<function _or_policy at 0x7e64dc2c84a0>, policies=[functools.partial(<function lambda_auto_wrap_policy at 0x7e64dc29be20>, lambda_fn=<function get_fsdp_wrap_policy.<locals>.lambda_policy_fn at 0x7e63661714e0>), functools.partial(<function transformer_auto_wrap_policy at 0x7e64dc2c8360>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])[36m(WorkerDict pid=2839719)[0m
|
| 499 |
+
[36m(WorkerDict pid=2841587)[0m
|
| 500 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 501 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 502 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 503 |
+
[36m(WorkerDict pid=2839719)[0m
|
| 504 |
+
[36m(WorkerDict pid=2839719)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 505 |
+
[36m(WorkerDict pid=2839719)[0m warnings.warn(
|
| 506 |
+
[33m(raylet)[0m [2026-06-12 12:09:38,824 E 2832550 2832550] (raylet) node_manager.cc:3303: 1 Workers (tasks / actors) killed due to memory pressure (OOM), 0 Workers crashed due to other reasons at node (ID: 49eb9cdf575181869c147cbd458d0f76c939299db90454b4d5aa27eb, IP: 172.20.115.25) over the last time period. To see more information about the Workers killed on this node, use `ray logs raylet.out -ip 172.20.115.25`
|
| 507 |
+
[33m(raylet)[0m
|
| 508 |
+
[33m(raylet)[0m Refer to the documentation on how to address the out of memory issue: https://docs.ray.io/en/latest/ray-core/scheduling/ray-oom-prevention.html. Consider provisioning more memory on this node or reducing task parallelism by requesting more CPUs per task. To adjust the kill threshold, set the environment variable `RAY_memory_usage_threshold` when starting Ray. To disable worker killing, set the environment variable `RAY_memory_monitor_refresh_ms` to zero.
|
| 509 |
+
[36m(WorkerDict pid=2841587)[0m /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:690: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
|
| 510 |
+
[36m(WorkerDict pid=2841587)[0m warnings.warn(
|
| 511 |
+
|
| 512 |
+
[36m(WorkerDict pid=2841587)[0m Actor use_remove_padding=False
|
| 513 |
+
[36m(WorkerDict pid=2839719)[0m Total steps: 472, num_warmup_steps: 0
|
| 514 |
+
[36m(WorkerDict pid=2839719)[0m Actor use_remove_padding=False
|
| 515 |
+
[36m(WorkerDict pid=2839719)[0m Before building vllm rollout, memory allocated (GB): 7.096388816833496, memory reserved (GB): 12.259765625
|
| 516 |
+
[36m(WorkerDict pid=2839719)[0m WARNING 06-12 12:01:58 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x7e62a0c00f50>
|
| 517 |
+
[36m(WorkerDict pid=2841587)[0m wrap_policy: functools.partial(<function _or_policy at 0x7bba8c2c44a0>, policies=[functools.partial(<function lambda_auto_wrap_policy at 0x7bba8c297e20>, lambda_fn=<function get_fsdp_wrap_policy.<locals>.lambda_policy_fn at 0x7bb9b43e3d80>), functools.partial(<function transformer_auto_wrap_policy at 0x7bba8c2c4360>, transformer_layer_cls={<class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLVisionBlock'>, <class 'transformers.models.qwen2_5_vl.modeling_qwen2_5_vl.Qwen2_5_VLDecoderLayer'>})])
|
| 518 |
+
[36m(WorkerDict pid=2841587)[0m Total steps: 472, num_warmup_steps: 0
|
| 519 |
+
[36m(WorkerDict pid=2841587)[0m Actor use_remove_padding=False
|
| 520 |
+
[36m(WorkerDict pid=2839719)[0m WARNING 06-12 12:02:03 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 521 |
+
[36m(WorkerDict pid=2841587)[0m WARNING 06-12 12:01:58 [utils.py:2522] Methods determine_num_available_blocks,device_config,get_cache_block_size_bytes,initialize_cache not implemented in <vllm.v1.worker.gpu_worker.Worker object at 0x7bb84b9fb090>
|
| 522 |
+
[36m(WorkerDict pid=2839719)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 523 |
+
[36m(WorkerDict pid=2839719)[0m After building vllm rollout, memory allocated (GB): 13.632704257965088, memory reserved (GB): 15.044921875
|
| 524 |
+
[36m(WorkerDict pid=2839719)[0m After building sharding manager, memory allocated (GB): 13.632704257965088, memory reserved (GB): 15.044921875
|
| 525 |
+
[36m(WorkerDict pid=2841587)[0m WARNING 06-12 12:02:03 [topk_topp_sampler.py:69] FlashInfer is not available. Falling back to the PyTorch-native implementation of top-p & top-k sampling. For the best performance, please install FlashInfer.
|
| 526 |
+
[36m(TaskRunner pid=2835960)[0m Using LocalLogger is deprecated. The constructor API will change
|
| 527 |
+
[36m(TaskRunner pid=2835960)[0m Found checkpoint: %s /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472
|
| 528 |
+
[36m(TaskRunner pid=2835960)[0m Load from checkpoint folder: /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472
|
| 529 |
+
[36m(TaskRunner pid=2835960)[0m Setting global step to 472
|
| 530 |
+
[36m(TaskRunner pid=2835960)[0m Resuming from /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472
|
| 531 |
+
[36m(WorkerDict pid=2839719)[0m [rank-0]: Loading from /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/model_world_size_2_rank_0.pt and /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/optim_world_size_2_rank_0.pt and /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/extra_state_world_size_2_rank_0.pt
|
| 532 |
+
[36m(TaskRunner pid=2835960)[0m test_gen_batch meta info: {'eos_token_id': 151645, 'pad_token_id': 151643, 'recompute_log_prob': False, 'do_sample': False, 'validate': True}
|
| 533 |
+
[36m(WorkerDict pid=2841587)[0m kwargs: {'n': 5, 'logprobs': 0, 'max_tokens': 1024, 'detokenize': False, 'temperature': 1.0, 'top_k': -1, 'top_p': 1, 'ignore_eos': False}
|
| 534 |
+
[36m(WorkerDict pid=2841587)[0m [rank-1]: Loading from /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/model_world_size_2_rank_1.pt and /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/optim_world_size_2_rank_1.pt and /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472/actor/extra_state_world_size_2_rank_1.pt
|
| 535 |
+
Error executing job with overrides: ['algorithm.adv_estimator=grpo', 'data.train_files=./data/rats_uni_processed/train_quarter.parquet', 'data.val_files=./data/rats_uni_processed/test_full.parquet', 'data.train_batch_size=16', 'data.max_prompt_length=1024', 'data.max_response_length=1024', 'data.filter_overlong_prompts=True', 'data.truncation=error', 'data.image_key=images', 'actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct', 'actor_rollout_ref.model.lora_rank=16', 'actor_rollout_ref.model.lora_alpha=16', 'actor_rollout_ref.model.lora_dropout=0.0', "actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'", 'actor_rollout_ref.actor.optim.lr=1e-4', 'actor_rollout_ref.model.use_remove_padding=False', 'actor_rollout_ref.actor.ppo_mini_batch_size=16', 'actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2', 'actor_rollout_ref.actor.use_kl_loss=True', 'actor_rollout_ref.actor.kl_loss_coef=0.001', 'actor_rollout_ref.actor.kl_loss_type=low_var_kl', 'actor_rollout_ref.model.enable_gradient_checkpointing=True', 'actor_rollout_ref.actor.fsdp_config.param_offload=False', 'actor_rollout_ref.actor.fsdp_config.optimizer_offload=False', 'actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.rollout.tensor_model_parallel_size=2', 'actor_rollout_ref.rollout.name=vllm', 'actor_rollout_ref.rollout.gpu_memory_utilization=0.4', 'actor_rollout_ref.rollout.enable_chunked_prefill=False', 'actor_rollout_ref.rollout.enforce_eager=False', 'actor_rollout_ref.rollout.free_cache_engine=False', 'actor_rollout_ref.rollout.n=5', 'actor_rollout_ref.rollout.val_kwargs.do_sample=False', 'actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8', 'actor_rollout_ref.ref.fsdp_config.param_offload=True', 'algorithm.kl_ctrl.kl_coef=0.001', 'trainer.critic_warmup=0', "trainer.logger=['console']", 'trainer.project_name=anomseer', 'trainer.experiment_name=anomseer_rats_uni_2gpu', 'trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu', 'trainer.n_gpus_per_node=2', 'trainer.nnodes=1', 'trainer.save_freq=500', 'trainer.test_freq=10', 'trainer.val_only=True', 'trainer.val_before_train=True', 'trainer.total_epochs=1', 'ts.use_sem_orth=True', 'ts.adv_mix=0.3', 'ts.similarity_method=ot', 'ts.ot_eps=0.08', 'ts.ot_n_iter=50']
|
| 536 |
+
Traceback (most recent call last):
|
| 537 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 68, in main
|
| 538 |
+
ray.get(runner.run.remote(config))
|
| 539 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/auto_init_hook.py", line 22, in auto_init_wrapper
|
| 540 |
+
return fn(*args, **kwargs)
|
| 541 |
+
^^^^^^^^^^^^^^^^^^^
|
| 542 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/client_mode_hook.py", line 107, in wrapper
|
| 543 |
+
return func(*args, **kwargs)
|
| 544 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 545 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 2980, in get
|
| 546 |
+
values, debugger_breakpoint = worker.get_objects(
|
| 547 |
+
^^^^^^^^^^^^^^^^^^^
|
| 548 |
+
File "/home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/_private/worker.py", line 1023, in get_objects
|
| 549 |
+
raise value.as_instanceof_cause()
|
| 550 |
+
ray.exceptions.RayTaskError(OutOfMemoryError): [36mray::TaskRunner.run()[39m (pid=2835960, ip=172.20.115.25, actor_id=4827f4aae735b6de5961046b01000000, repr=<main_ppo.TaskRunner object at 0x735c213ffa50>)
|
| 551 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 552 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 553 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/main_ppo.py", line 173, in run
|
| 554 |
+
trainer.fit()
|
| 555 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/ppo/ray_trainer.py", line 2196, in fit
|
| 556 |
+
val_metrics = self._validate()
|
| 557 |
+
^^^^^^^^^^^^^^^^
|
| 558 |
+
File "/mnt/share01/sqk/AnomSeer/verl/trainer/ppo/ray_trainer.py", line 1840, in _validate
|
| 559 |
+
test_output_gen_batch_padded = self.actor_rollout_wg.generate_sequences(test_gen_batch_padded)
|
| 560 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 561 |
+
File "/mnt/share01/sqk/AnomSeer/verl/single_controller/ray/base.py", line 43, in func
|
| 562 |
+
output = ray.get(output)
|
| 563 |
+
^^^^^^^^^^^^^^^
|
| 564 |
+
^^^^^^^^^^^^^^^^^^^
|
| 565 |
+
^^^^^^^^^^^^^^^^^^^^^
|
| 566 |
+
^^^^^^^^^^^^^^^^^^^
|
| 567 |
+
ray.exceptions.OutOfMemoryError: 1 worker(s) were killed due to the node running low on memory. Memory on the node (IP: 172.20.115.25, ID: 49eb9cdf575181869c147cbd458d0f76c939299db90454b4d5aa27eb) was 239.55GB / 251.21GB (0.953616), which exceeds the memory usage threshold of 0.950000; Object store memory usage: [- objects spillable: 0; - bytes spillable: 0; - objects unsealed: 0; - bytes unsealed: 0; - objects in use: 1; - bytes in use: 20367164421; - objects evictable: 0; - bytes evictable: 0; ; - objects created by worker: 1; - bytes created by worker: 20367164421; - objects restored: 0; - bytes restored: 0; - objects received: 0; - bytes received: 0; - objects errored: 0; - bytes errored: 0; ; Eviction Stats:; (global lru) capacity: 72791529062; (global lru) used: 0%; (global lru) num objects: 0; (global lru) num evictions: 0; (global lru) bytes evicted: 0]; Ray killed 1 worker(s) based on the killing policy: [(Actor(4f145c2210673ecf4c40aa0d01000000): job ID=01000000, lease ID=03000000d21587b00dcc5a97a787fed0e40fafc5b956e0f280f921aed06b8784, task name=R4qskTWorkerDict_0:1:WorkerDict.__init__, required resources={GPU_group_218e351d7bf7afdb768364c8f98301000000: 1, bundle_group_1_218e351d7bf7afdb768364c8f98301000000: 0.001, CPU_group_218e351d7bf7afdb768364c8f98301000000: 1, GPU_group_1_218e351d7bf7afdb768364c8f98301000000: 1, CPU_group_1_218e351d7bf7afdb768364c8f98301000000: 1, bundle_group_218e351d7bf7afdb768364c8f98301000000: 0.001}, pid=2841587, actual memory used=14.59GB, worker ID=1af15360d55724f19e6b7f6b7eaf205cab912f858e2e6ef3b4745534)]; To see more information about memory usage on this node, use `ray logs raylet.out -ip 172.20.115.25`; Top 10 memory users: PID MEM(GB) COMMAND, 2835960 78.85 ray::TaskRunner.run, 2839719 50.54 ray::WorkerDict.actor_rollout_generate_sequences, 2874386 47.90 ray::TaskRunner.run, 2841587 14.59 ray::WorkerDict, 2349684 0.55 /home/suiqk/.vscode-server/cli/servers/Stable-1b50d58d73426c9171299ec4037d01365d995b78/server/node /..., 2830946 0.46 /home/suiqk/anaconda3/envs/scalerag-ts-v4/bin/python -m verl.trainer.main_ppo algorithm.adv_estimato..., 2133445 0.41 /home/suiqk/.vscode-server/cli/servers/Stable-1b50d58d73426c9171299ec4037d01365d995b78/server/node -..., 2133423 0.35 /home/suiqk/.vscode-server/cli/servers/Stable-1b50d58d73426c9171299ec4037d01365d995b78/server/node /..., 2832066 0.31 /home/suiqk/anaconda3/envs/scalerag-ts-v4/lib/python3.11/site-packages/ray/core/src/ray/gcs/gcs_serv..., 1402406 0.18 /usr/bin/python3 /usr/bin/update-manager --no-update --no-focus-on-map, Refer to the documentation on how to address the out of memory issue: https://docs.ray.io/en/latest/ray-core/scheduling/ray-oom-prevention.html. Consider provisioning more memory on this node or reducing task parallelism by requesting more CPUs per task. Set max_restarts and max_task_retries to enable retry when the task crashes due to OOM. To adjust the kill threshold, set the environment variable `RAY_memory_usage_threshold` when starting Ray. To disable worker killing, set the environment variable `RAY_memory_monitor_refresh_ms` to zero.
|
| 568 |
+
|
| 569 |
+
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
|
logs/rats_eval_full2.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/rats_eval_full2_metrics.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoint": "checkpoints/anomseer/anomseer_rats_uni_2gpu/global_step_472",
|
| 3 |
+
"val_file": "data/rats_uni_processed/test_full.parquet",
|
| 4 |
+
"decoding": "greedy (do_sample=False, val_kwargs.n=1)",
|
| 5 |
+
"num_total_samples": 6034,
|
| 6 |
+
"num_valid_samples": 6034,
|
| 7 |
+
"num_invalid_predictions": 0,
|
| 8 |
+
"type_accuracy": 0.14368578057673184,
|
| 9 |
+
"type_precision_macro": 0.07011846062585,
|
| 10 |
+
"type_recall_macro": 0.15190385921696692,
|
| 11 |
+
"type_f1_macro": 0.06658531363554307,
|
| 12 |
+
"binary_accuracy": 0.8367583692409678,
|
| 13 |
+
"binary_precision_macro": 0.7156772435960252,
|
| 14 |
+
"binary_recall_macro": 0.6171428305303734,
|
| 15 |
+
"binary_f1_macro": 0.6403655346950877,
|
| 16 |
+
"reward_score": 0.2986244636942485,
|
| 17 |
+
"mean_fmt_score": 1.0
|
| 18 |
+
}
|
logs/rats_train_lora_20260611_174333.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
multimodal_data_processing/anom.py
ADDED
|
@@ -0,0 +1,596 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# -*- coding: utf-8 -*-
|
| 2 |
+
|
| 3 |
+
import os
|
| 4 |
+
import re
|
| 5 |
+
import json
|
| 6 |
+
import glob
|
| 7 |
+
import argparse
|
| 8 |
+
import pickle
|
| 9 |
+
from typing import List, Tuple, Optional, Dict
|
| 10 |
+
|
| 11 |
+
import numpy as np
|
| 12 |
+
import pandas as pd
|
| 13 |
+
import stumpy
|
| 14 |
+
from PIL import Image
|
| 15 |
+
from datasets import Dataset
|
| 16 |
+
from scipy.signal import savgol_filter
|
| 17 |
+
from scipy.stats import zscore
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
def _dominant_period(x: np.ndarray) -> float:
|
| 21 |
+
"""Computes the dominant period of a time series using FFT."""
|
| 22 |
+
x = np.asarray(x, np.float32)
|
| 23 |
+
n = x.size
|
| 24 |
+
if n < 8: return float('inf')
|
| 25 |
+
xdm = x - x.mean()
|
| 26 |
+
Y = np.fft.rfft(xdm)
|
| 27 |
+
P = np.abs(Y)**2
|
| 28 |
+
if P.size > 0: P[0] = 0.0
|
| 29 |
+
freqs = np.fft.rfftfreq(n, d=1.0)
|
| 30 |
+
if P.size <= 1: return float('inf')
|
| 31 |
+
k = int(np.argmax(P))
|
| 32 |
+
f = float(freqs[k])
|
| 33 |
+
return float(1.0 / f) if f > 0 else float('inf')
|
| 34 |
+
|
| 35 |
+
def vector_to_intervals(vec: np.ndarray) -> List[List[int]]:
|
| 36 |
+
"""Converts a binary (0/1) vector to a list of [start, end) intervals."""
|
| 37 |
+
vec = np.asarray(vec).astype(int).flatten()
|
| 38 |
+
intervals = []
|
| 39 |
+
in_seg = False
|
| 40 |
+
start = 0
|
| 41 |
+
for i, v in enumerate(vec):
|
| 42 |
+
if v == 1 and not in_seg:
|
| 43 |
+
start = i
|
| 44 |
+
in_seg = True
|
| 45 |
+
elif v == 0 and in_seg:
|
| 46 |
+
intervals.append([start, i])
|
| 47 |
+
in_seg = False
|
| 48 |
+
if in_seg:
|
| 49 |
+
intervals.append([start, len(vec)])
|
| 50 |
+
return intervals
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
|
| 54 |
+
def _try_load_npy(path: str) -> Optional[np.ndarray]:
|
| 55 |
+
try:
|
| 56 |
+
if os.path.isfile(path):
|
| 57 |
+
return np.load(path, allow_pickle=True)
|
| 58 |
+
except Exception:
|
| 59 |
+
return None
|
| 60 |
+
|
| 61 |
+
def _find_series_and_gt(root_task_dir: str, split: str, idx: str) -> Tuple[Optional[np.ndarray], List[List[int]]]:
|
| 62 |
+
"""Finds raw series and ground truth intervals from common directory patterns."""
|
| 63 |
+
series_path = os.path.join(root_task_dir, split, "series", f"{idx}.npy")
|
| 64 |
+
labels_path = os.path.join(root_task_dir, split, "labels", f"{idx}.npy")
|
| 65 |
+
|
| 66 |
+
ts = _try_load_npy(series_path)
|
| 67 |
+
gt_vec = _try_load_npy(labels_path)
|
| 68 |
+
|
| 69 |
+
intervals = []
|
| 70 |
+
if gt_vec is not None:
|
| 71 |
+
try:
|
| 72 |
+
intervals = vector_to_intervals(gt_vec.astype(int))
|
| 73 |
+
except Exception:
|
| 74 |
+
pass
|
| 75 |
+
|
| 76 |
+
return ts, intervals
|
| 77 |
+
|
| 78 |
+
def _parse_idx_from_png(png_path: str) -> str:
|
| 79 |
+
"""Extracts the numerical index from a PNG filename."""
|
| 80 |
+
base = os.path.basename(png_path)
|
| 81 |
+
m = re.search(r"(\d+)", base)
|
| 82 |
+
return m.group(1) if m else os.path.splitext(base)[0]
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
|
| 86 |
+
|
| 87 |
+
def detect_range_anomalies(ts: np.ndarray, k: float = 3.0) -> Tuple[List[List[int]], str, Dict]:
|
| 88 |
+
"""Uses k-sigma rule for range anomalies and generates a CoT-style text."""
|
| 89 |
+
if ts.size < 2:
|
| 90 |
+
return [], "Series too short.", {}
|
| 91 |
+
|
| 92 |
+
mu, std = np.mean(ts), np.std(ts)
|
| 93 |
+
if std < 1e-6:
|
| 94 |
+
return [], f"The series is constant (value={mu:.2f}), so no range anomalies can be detected.", {}
|
| 95 |
+
|
| 96 |
+
upper, lower = mu + k * std, mu - k * std
|
| 97 |
+
is_anomaly = (ts > upper) | (ts < lower)
|
| 98 |
+
intervals = vector_to_intervals(is_anomaly)
|
| 99 |
+
|
| 100 |
+
analysis_text = (
|
| 101 |
+
f"To find range anomalies, I first calculated the global statistics: "
|
| 102 |
+
f"the mean is {mu:.3f} and the standard deviation is {std:.3f}. "
|
| 103 |
+
f"This establishes a normal range of [{lower:.3f}, {upper:.3f}] using a {k}-sigma rule. "
|
| 104 |
+
f"Scanning the series for points outside this boundary resulted in finding {len(intervals)} anomalous interval(s)."
|
| 105 |
+
)
|
| 106 |
+
metrics = {"range_mean": mu, "range_std": std, "range_upper": upper, "range_lower": lower}
|
| 107 |
+
return intervals, analysis_text, metrics
|
| 108 |
+
|
| 109 |
+
def detect_trend_anomalies(ts: np.ndarray, window: int = 21, k: float = 3.0) -> Tuple[List[List[int]], str, Dict]:
|
| 110 |
+
"""Uses smoothed gradient for trend anomalies and generates a CoT-style text."""
|
| 111 |
+
if ts.size < window * 2:
|
| 112 |
+
return [], "Series too short for trend analysis.", {}
|
| 113 |
+
|
| 114 |
+
smooth_ts = savgol_filter(ts, window_length=window, polyorder=2)
|
| 115 |
+
gradient = np.gradient(smooth_ts)
|
| 116 |
+
grad_mu, grad_std = np.mean(gradient), np.std(gradient)
|
| 117 |
+
if grad_std < 1e-6:
|
| 118 |
+
return [], f"The trend appears to be constant (gradient={grad_mu:.4f}), so no significant changes were detected.", {}
|
| 119 |
+
|
| 120 |
+
threshold = k * grad_std
|
| 121 |
+
is_anomaly = np.abs(gradient - grad_mu) > threshold
|
| 122 |
+
intervals = vector_to_intervals(is_anomaly)
|
| 123 |
+
|
| 124 |
+
analysis_text = (
|
| 125 |
+
f"For trend analysis, I focused on the rate of change. After smoothing the series with a window of {window}, "
|
| 126 |
+
f"I computed its gradient. The normal gradient fluctuates around a mean of {grad_mu:.4f} with a std of {grad_std:.4f}. "
|
| 127 |
+
f"A significant trend shift should cause a gradient deviation larger than the {k}-sigma threshold of {threshold:.4f}. "
|
| 128 |
+
f"This check identified {len(intervals)} interval(s) meeting the condition."
|
| 129 |
+
)
|
| 130 |
+
metrics = {"trend_grad_mean": grad_mu, "trend_grad_std": grad_std, "trend_grad_threshold": threshold}
|
| 131 |
+
return intervals, analysis_text, metrics
|
| 132 |
+
|
| 133 |
+
def detect_point_anomalies(ts: np.ndarray, window: int = 50, k: float = 3.5) -> Tuple[List[List[int]], str, Dict]:
|
| 134 |
+
"""Uses Matrix Profile (STUMPY) for point anomalies and generates a CoT-style text."""
|
| 135 |
+
if ts.size < window * 2:
|
| 136 |
+
return [], "Series too short for discord detection.", {}
|
| 137 |
+
|
| 138 |
+
|
| 139 |
+
mp = stumpy.stump(ts, m=window, ignore_trivial=True)
|
| 140 |
+
|
| 141 |
+
|
| 142 |
+
|
| 143 |
+
raw_mp_distances = mp[:, 0]
|
| 144 |
+
mp_distances = np.zeros(raw_mp_distances.shape, dtype=np.float64)
|
| 145 |
+
for i, val in enumerate(raw_mp_distances):
|
| 146 |
+
try:
|
| 147 |
+
float_val = float(val)
|
| 148 |
+
mp_distances[i] = float_val
|
| 149 |
+
except (ValueError, TypeError):
|
| 150 |
+
mp_distances[i] = 0.0
|
| 151 |
+
|
| 152 |
+
finite_mask = np.isfinite(mp_distances)
|
| 153 |
+
if np.any(finite_mask):
|
| 154 |
+
max_finite_val = np.max(mp_distances[finite_mask])
|
| 155 |
+
mp_distances[~finite_mask] = max_finite_val
|
| 156 |
+
else:
|
| 157 |
+
return [], "Matrix Profile computation resulted in non-finite values only.", {}
|
| 158 |
+
|
| 159 |
+
mp_zscores = zscore(mp_distances)
|
| 160 |
+
discord_idx = np.argmax(mp_distances)
|
| 161 |
+
max_zscore = mp_zscores[discord_idx]
|
| 162 |
+
|
| 163 |
+
intervals = []
|
| 164 |
+
if max_zscore > k:
|
| 165 |
+
intervals = [[int(discord_idx), int(discord_idx + window)]]
|
| 166 |
+
|
| 167 |
+
analysis_text = (
|
| 168 |
+
f"To detect contextual point anomalies, I used the Matrix Profile (window={window}, ignoring trivial matches) to find the most unusual subsequence (discord). "
|
| 169 |
+
f"The computation revealed that the highest discord score is {mp_distances[discord_idx]:.2f}, located at index {discord_idx}. "
|
| 170 |
+
f"This score corresponds to a z-score of {max_zscore:.2f}, which is above my threshold of {k}, indicating a significant anomaly. "
|
| 171 |
+
f"Thus, {len(intervals)} anomaly was identified."
|
| 172 |
+
)
|
| 173 |
+
metrics = {"point_discord_idx": int(discord_idx), "point_discord_zscore": float(max_zscore)}
|
| 174 |
+
return intervals, analysis_text, metrics
|
| 175 |
+
|
| 176 |
+
def detect_freq_anomalies(ts: np.ndarray, window: int = 100, k: float = 3.0) -> Tuple[List[List[int]], str, Dict]:
|
| 177 |
+
"""Uses dominant period changes for frequency anomalies and generates a CoT-style text."""
|
| 178 |
+
if ts.size < window * 3:
|
| 179 |
+
return [], "Series too short for frequency analysis.", {}
|
| 180 |
+
|
| 181 |
+
periods = [_dominant_period(ts[i : i + window]) for i in range(ts.size - window)]
|
| 182 |
+
periods = np.array([p if np.isfinite(p) else -1 for p in periods])
|
| 183 |
+
valid_periods = periods[periods != -1]
|
| 184 |
+
|
| 185 |
+
if valid_periods.size < 10:
|
| 186 |
+
return [], "Could not determine a stable dominant period.", {}
|
| 187 |
+
|
| 188 |
+
per_mu, per_std = np.mean(valid_periods), np.std(valid_periods)
|
| 189 |
+
if per_std < 1.0: # If period is very stable
|
| 190 |
+
return [], f"The dominant period is highly stable around {per_mu:.2f}, so no frequency anomalies detected.", {}
|
| 191 |
+
|
| 192 |
+
threshold_upper = per_mu + k * per_std
|
| 193 |
+
threshold_lower = per_mu - k * per_std
|
| 194 |
+
is_anomaly_periods = (periods > threshold_upper) | ((periods < threshold_lower) & (periods != -1))
|
| 195 |
+
|
| 196 |
+
is_anomaly_ts = np.zeros_like(ts, dtype=bool)
|
| 197 |
+
for i, is_anom in enumerate(is_anomaly_periods):
|
| 198 |
+
if is_anom:
|
| 199 |
+
is_anomaly_ts[i : i + window] = True
|
| 200 |
+
|
| 201 |
+
intervals = vector_to_intervals(is_anomaly_ts)
|
| 202 |
+
|
| 203 |
+
analysis_text = (
|
| 204 |
+
f"To find frequency anomalies, I calculated the dominant period over a sliding window of size {window}. "
|
| 205 |
+
f"The typical period is around {per_mu:.2f} with a std of {per_std:.2f}. "
|
| 206 |
+
f"I'm looking for regions where the period significantly deviates from the normal range of [{threshold_lower:.2f}, {threshold_upper:.2f}]. "
|
| 207 |
+
f"This analysis pointed to {len(intervals)} interval(s) with clear frequency shifts."
|
| 208 |
+
)
|
| 209 |
+
metrics = {"freq_period_mean": per_mu, "freq_period_std": per_std}
|
| 210 |
+
return intervals, analysis_text, metrics
|
| 211 |
+
|
| 212 |
+
|
| 213 |
+
def _json_default(o):
|
| 214 |
+
import numpy as _np
|
| 215 |
+
if isinstance(o, (_np.integer,)):
|
| 216 |
+
return int(o)
|
| 217 |
+
if isinstance(o, (_np.floating,)):
|
| 218 |
+
return float(o)
|
| 219 |
+
if isinstance(o, _np.ndarray):
|
| 220 |
+
return o.tolist()
|
| 221 |
+
return str(o)
|
| 222 |
+
|
| 223 |
+
def _intervals_str(intervals):
|
| 224 |
+
if not intervals:
|
| 225 |
+
return "[]"
|
| 226 |
+
return ", ".join([f"[{int(s)}, {int(e)}]" for s, e in intervals])
|
| 227 |
+
|
| 228 |
+
def _normalize_class(name: str) -> str:
|
| 229 |
+
m = (name or "").lower()
|
| 230 |
+
if m in ["global","global point","out-of-range","range","noisy-range"]:
|
| 231 |
+
return "global point"
|
| 232 |
+
if m in ["contextual","contextual point","point","noisy-point"]:
|
| 233 |
+
return "contextual point"
|
| 234 |
+
if m in ["trend","trend shift","noisy-trend","flat-trend"]:
|
| 235 |
+
return "trend"
|
| 236 |
+
if m in ["seasonal","frequency","seasonal/frequency deviation","freq","noisy-freq"]:
|
| 237 |
+
return "seasonal"
|
| 238 |
+
if m in ["shapelet","subsequence","shapelet/subsequence"]:
|
| 239 |
+
return "shapelet"
|
| 240 |
+
return "normal"
|
| 241 |
+
|
| 242 |
+
|
| 243 |
+
def _get_expert_reasoning_flow(ts: np.ndarray, task_hint: str) -> Tuple[str, callable]:
|
| 244 |
+
"""
|
| 245 |
+
Simulates an expert's diagnostic process to select the right tool.
|
| 246 |
+
Returns a reasoning text and the selected detection function.
|
| 247 |
+
"""
|
| 248 |
+
# --- Step 1: Global Scan
|
| 249 |
+
global_mu, global_std = np.mean(ts), np.std(ts)
|
| 250 |
+
max_zscore = 0
|
| 251 |
+
if global_std > 1e-6:
|
| 252 |
+
z_scores = np.abs((ts - global_mu) / global_std)
|
| 253 |
+
max_zscore = np.max(z_scores)
|
| 254 |
+
|
| 255 |
+
if max_zscore > 5.0: # A very high Z-score suggests a simple range anomaly is likely.
|
| 256 |
+
reasoning = (
|
| 257 |
+
"My initial check reveals extreme values. The global mean is "
|
| 258 |
+
f"{global_mu:.3f} and std is {global_std:.3f}, but some points have a z-score as high as {max_zscore:.2f}. "
|
| 259 |
+
"This strongly suggests a range-based anomaly. I will now apply a k-sigma rule to formalize this."
|
| 260 |
+
)
|
| 261 |
+
return reasoning, detect_range_anomalies
|
| 262 |
+
|
| 263 |
+
# --- Step 2: Structural Scan
|
| 264 |
+
|
| 265 |
+
# Check for stable trend
|
| 266 |
+
gradient = np.gradient(ts)
|
| 267 |
+
grad_mu, grad_std = np.mean(gradient), np.std(gradient)
|
| 268 |
+
# A low gradient std relative to the overall data std might indicate a stable trend
|
| 269 |
+
is_trend_stable = (grad_std / global_std) < 0.1 if global_std > 1e-6 else True
|
| 270 |
+
|
| 271 |
+
# Check for stable frequency
|
| 272 |
+
dominant_p = _dominant_period(ts)
|
| 273 |
+
is_freq_stable = dominant_p != float('inf') and dominant_p > 8 # has a detectable period
|
| 274 |
+
|
| 275 |
+
# Decision based on structure and task hint
|
| 276 |
+
task_lc = task_hint.lower()
|
| 277 |
+
if 'trend' in task_lc:
|
| 278 |
+
reasoning = (
|
| 279 |
+
f"The global values seem normal (max z-score={max_zscore:.2f}), so I'll check the trend. "
|
| 280 |
+
"The gradient of the series appears unstable. This suggests a potential trend anomaly. "
|
| 281 |
+
"I will analyze the smoothed gradient to confirm any significant shifts."
|
| 282 |
+
)
|
| 283 |
+
return reasoning, detect_trend_anomalies
|
| 284 |
+
|
| 285 |
+
if 'freq' in task_lc:
|
| 286 |
+
reasoning = (
|
| 287 |
+
f"Global values and trend seem stable. However, the signal appears periodic. "
|
| 288 |
+
"An unstable period can indicate a frequency anomaly. "
|
| 289 |
+
"I will use a sliding window analysis to check for significant changes in the dominant period."
|
| 290 |
+
)
|
| 291 |
+
return reasoning, detect_freq_anomalies
|
| 292 |
+
|
| 293 |
+
# --- Step 3: Pattern Scan (If all else seems normal, look for unique patterns) ---
|
| 294 |
+
# Corresponds to Matrix Profile logic. This is the default for 'point' or when other checks fail.
|
| 295 |
+
reasoning = (
|
| 296 |
+
f"The series does not exhibit obvious global outliers (max z-score={max_zscore:.2f}) or clear structural instability. "
|
| 297 |
+
"The anomalies are likely subtle and contextual. This requires a pattern-based approach. "
|
| 298 |
+
"I will use the Matrix Profile to find the most dissimilar subsequence (a discord), which is the standard method for such cases."
|
| 299 |
+
)
|
| 300 |
+
return reasoning, detect_point_anomalies
|
| 301 |
+
|
| 302 |
+
|
| 303 |
+
def _link_gt_feature(gt_intervals: List[List[int]], class_name: str) -> str:
|
| 304 |
+
iv_str = _intervals_str(gt_intervals)
|
| 305 |
+
if iv_str == "[]":
|
| 306 |
+
return f"I did not observe any {class_name} anomaly in the series. "
|
| 307 |
+
|
| 308 |
+
if class_name == "global point":
|
| 309 |
+
return f"I observed that the values within {iv_str} exhibit clear out-of-range behavior, with sharp spikes deviating from the global distribution. "
|
| 310 |
+
if class_name == "contextual point":
|
| 311 |
+
return f"I observed that the subsequence {iv_str} appears inconsistent with its local temporal context, breaking the continuity of surrounding patterns. "
|
| 312 |
+
if class_name == "trend":
|
| 313 |
+
return f"I observed that the segment within {iv_str} shows a clear trend shift, with the long-term slope undergoing a marked change. "
|
| 314 |
+
if class_name == "seasonal":
|
| 315 |
+
return f"I observed that the oscillations within {iv_str} display frequency deviation, with periodic structure misaligned from the baseline cycle. "
|
| 316 |
+
if class_name == "shapelet":
|
| 317 |
+
return f"I observed that the subsequence within {iv_str} deviates in waveform shape, differing notably from typical local motifs. "
|
| 318 |
+
return f"I observed that the segment {iv_str} is annotated as {class_name}. "
|
| 319 |
+
|
| 320 |
+
|
| 321 |
+
def build_prompt_and_expcot(ts: Optional[np.ndarray],
|
| 322 |
+
task: str,
|
| 323 |
+
L: int,
|
| 324 |
+
gt_intervals: List[List[int]],
|
| 325 |
+
gt_type: str) -> Tuple[str, str, Dict]:
|
| 326 |
+
|
| 327 |
+
prompt = (
|
| 328 |
+
"<image>\n"
|
| 329 |
+
f"You are a time series analysis expert. A time series plot of length {L} is provided. "
|
| 330 |
+
"Identify anomalous intervals along the x-axis and infer the most plausible anomaly type from "
|
| 331 |
+
"[\"contextual point\", \"global point\", \"seasonal\", \"trend\", \"shapelet\"].\n\n"
|
| 332 |
+
"Begin detailed reasoning inside <think>...</think>.\n"
|
| 333 |
+
"Then output:\n"
|
| 334 |
+
"<answer>[[start, end], ...]</answer>\n"
|
| 335 |
+
"<class>one of {contextual point, global point, seasonal, trend, shapelet, normal}</class>\n"
|
| 336 |
+
"If no anomalies, return <answer>[]</answer> and <class>normal</class>.\n"
|
| 337 |
+
)
|
| 338 |
+
|
| 339 |
+
|
| 340 |
+
class_by_task = _normalize_class(task)
|
| 341 |
+
class_by_gt = _normalize_class(gt_type)
|
| 342 |
+
class_name = class_by_gt if class_by_gt != "normal" else class_by_task
|
| 343 |
+
|
| 344 |
+
if ts is None or ts.size < 20:
|
| 345 |
+
if gt_intervals:
|
| 346 |
+
expcot = (
|
| 347 |
+
f"\\textbf{{Observation}} — Ground truth marks { _intervals_str(gt_intervals) } "
|
| 348 |
+
f"as \\textit{{{class_name}}}, but raw series is unavailable/too short for verification.\n"
|
| 349 |
+
f"\\textbf{{Conclusion}} — We retain the GT label and intervals for supervision."
|
| 350 |
+
)
|
| 351 |
+
else:
|
| 352 |
+
expcot = (
|
| 353 |
+
"\\textbf{Observation} — Ground truth indicates no anomaly; raw series unavailable/too short.\n"
|
| 354 |
+
"\\textbf{Conclusion} — Treated as normal."
|
| 355 |
+
)
|
| 356 |
+
return prompt, expcot, {"intervals_gt": gt_intervals, "class": class_name}
|
| 357 |
+
|
| 358 |
+
|
| 359 |
+
x = ts if ts.ndim == 1 else ts.mean(axis=1)
|
| 360 |
+
pre = ""
|
| 361 |
+
if "noisy" in task.lower():
|
| 362 |
+
wl = max(5, min(31, (len(x)//4)|1))
|
| 363 |
+
x = savgol_filter(x, window_length=wl, polyorder=2)
|
| 364 |
+
pre = f"[Preprocess] Applied Savitzky–Golay denoising (win={wl}). "
|
| 365 |
+
|
| 366 |
+
initial_reasoning, selected_detector = _get_expert_reasoning_flow(x, task)
|
| 367 |
+
det_intervals, final_evidence, detection_metrics = selected_detector(x)
|
| 368 |
+
|
| 369 |
+
|
| 370 |
+
obs = (
|
| 371 |
+
f"Ground truth labels { _intervals_str(gt_intervals) } as \\textit{{{class_name}}}. "
|
| 372 |
+
f"{initial_reasoning}"
|
| 373 |
+
)
|
| 374 |
+
|
| 375 |
+
|
| 376 |
+
val = (
|
| 377 |
+
_link_gt_feature(gt_intervals, class_name)
|
| 378 |
+
+ final_evidence.replace("To find", "The analysis").replace("For", "The analysis")
|
| 379 |
+
)
|
| 380 |
+
|
| 381 |
+
|
| 382 |
+
|
| 383 |
+
if gt_intervals:
|
| 384 |
+
concl = (
|
| 385 |
+
f"We report the GT interval(s) { _intervals_str(gt_intervals) } "
|
| 386 |
+
f"as the final localization for supervision. "
|
| 387 |
+
f"For reference, the detector proposed { _intervals_str(det_intervals) }."
|
| 388 |
+
)
|
| 389 |
+
out_intervals = gt_intervals #
|
| 390 |
+
else:
|
| 391 |
+
concl = (
|
| 392 |
+
"No ground-truth anomalies are present; the series is treated as normal. "
|
| 393 |
+
f"For reference, the detector proposed { _intervals_str(det_intervals) }."
|
| 394 |
+
)
|
| 395 |
+
out_intervals = [] # normal
|
| 396 |
+
|
| 397 |
+
expcot = (
|
| 398 |
+
f"{pre}"
|
| 399 |
+
f"\\textbf{{Observation}} — {obs}\n"
|
| 400 |
+
f"\\textbf{{Reasoning \\& Validation}} — {val}\n"
|
| 401 |
+
f"\\textbf{{Conclusion}} — {concl}"
|
| 402 |
+
).strip()
|
| 403 |
+
|
| 404 |
+
|
| 405 |
+
return prompt, expcot, {
|
| 406 |
+
"intervals_gt": gt_intervals,
|
| 407 |
+
"intervals_pred": det_intervals,
|
| 408 |
+
"class": class_name,
|
| 409 |
+
**detection_metrics
|
| 410 |
+
}
|
| 411 |
+
|
| 412 |
+
# --- CORRECTED DATA LOADING HELPERS ---
|
| 413 |
+
def _load_split_pkl(task_dir: str, split: str) -> Tuple[Optional[list], Optional[list]]:
|
| 414 |
+
|
| 415 |
+
pkl_path = os.path.join(task_dir, split, "data.pkl")
|
| 416 |
+
if not os.path.isfile(pkl_path):
|
| 417 |
+
return None, None
|
| 418 |
+
try:
|
| 419 |
+
with open(pkl_path, "rb") as f:
|
| 420 |
+
obj = pickle.load(f)
|
| 421 |
+
series = obj.get("series", None)
|
| 422 |
+
anom = obj.get("anom", None)
|
| 423 |
+
return series, anom
|
| 424 |
+
except Exception:
|
| 425 |
+
return None, None
|
| 426 |
+
|
| 427 |
+
def _pick_index_from_png_idx(idx_str: str, n: int) -> Optional[int]:
|
| 428 |
+
|
| 429 |
+
try:
|
| 430 |
+
k = int(idx_str)
|
| 431 |
+
except Exception:
|
| 432 |
+
return None
|
| 433 |
+
for cand in (k-1, k):
|
| 434 |
+
if 0 <= cand < n:
|
| 435 |
+
return cand
|
| 436 |
+
return None
|
| 437 |
+
|
| 438 |
+
def _intervals_from_anom_entry(anom_entry) -> List[List[int]]:
|
| 439 |
+
|
| 440 |
+
intervals = []
|
| 441 |
+
if not anom_entry: return intervals
|
| 442 |
+
for ch_list in anom_entry:
|
| 443 |
+
for pair in ch_list:
|
| 444 |
+
if isinstance(pair, (list, tuple)) and len(pair) == 2:
|
| 445 |
+
s, e = int(pair[0]), int(pair[1])
|
| 446 |
+
if e > s: intervals.append([s, e])
|
| 447 |
+
intervals.sort(key=lambda z: (z[0], z[1]))
|
| 448 |
+
return intervals
|
| 449 |
+
|
| 450 |
+
def _parse_idx_from_png(png_path: str) -> str:
|
| 451 |
+
|
| 452 |
+
base = os.path.basename(png_path)
|
| 453 |
+
m = re.search(r"(\d+)", base)
|
| 454 |
+
return m.group(1) if m else os.path.splitext(base)[0]
|
| 455 |
+
|
| 456 |
+
def process_task(root_dir: str, task: str, out_dir: str) -> Tuple[str, Optional[str]]:
|
| 457 |
+
|
| 458 |
+
task_dir = os.path.join(root_dir, task)
|
| 459 |
+
results = {"train": [], "eval": []}
|
| 460 |
+
|
| 461 |
+
for split in ["train", "eval"]:
|
| 462 |
+
figs_dir = os.path.join(task_dir, split, "figs")
|
| 463 |
+
if not os.path.isdir(figs_dir):
|
| 464 |
+
print(f"Warning: Directory not found for {task}/{split}/figs. Skipping.")
|
| 465 |
+
continue
|
| 466 |
+
|
| 467 |
+
|
| 468 |
+
series_list, anom_list = _load_split_pkl(task_dir, split)
|
| 469 |
+
n_samples = len(series_list) if isinstance(series_list, list) else 0
|
| 470 |
+
use_pkl = (series_list is not None) and (anom_list is not None) and (n_samples > 0)
|
| 471 |
+
|
| 472 |
+
pngs = sorted(glob.glob(os.path.join(figs_dir, "*.png")))
|
| 473 |
+
for i, png_path in enumerate(pngs):
|
| 474 |
+
idx_str = _parse_idx_from_png(png_path)
|
| 475 |
+
|
| 476 |
+
ts, gt_intervals = None, []
|
| 477 |
+
|
| 478 |
+
|
| 479 |
+
if use_pkl:
|
| 480 |
+
pkl_i = _pick_index_from_png_idx(idx_str, n_samples)
|
| 481 |
+
if pkl_i is not None:
|
| 482 |
+
ts = np.asarray(series_list[pkl_i])
|
| 483 |
+
gt_intervals = _intervals_from_anom_entry(anom_list[pkl_i])
|
| 484 |
+
|
| 485 |
+
|
| 486 |
+
def map_task_to_anomaly(task_name: str) -> str:
|
| 487 |
+
if task_name in ["range", "noisy-range"]:
|
| 488 |
+
return "global"
|
| 489 |
+
elif task_name in ["freq", "noise-freq"]:
|
| 490 |
+
return "seasonal"
|
| 491 |
+
elif task_name in ["point", "noise-point"]:
|
| 492 |
+
return "contextual"
|
| 493 |
+
elif task_name in ["trend", "noise-trend"]:
|
| 494 |
+
return "trend"
|
| 495 |
+
else:
|
| 496 |
+
return "unknown"
|
| 497 |
+
L = len(ts) if ts is not None else 0
|
| 498 |
+
|
| 499 |
+
anomaly_type = map_task_to_anomaly(task)
|
| 500 |
+
if gt_intervals == []:
|
| 501 |
+
anomaly_type = 'normal'
|
| 502 |
+
gt_type = anomaly_type if anomaly_type else _normalize_class(task)
|
| 503 |
+
prompt_text, expcot, detection_metrics = build_prompt_and_expcot(ts, task, L, gt_intervals, gt_type)
|
| 504 |
+
|
| 505 |
+
image = Image.open(png_path).convert("RGBA")
|
| 506 |
+
|
| 507 |
+
|
| 508 |
+
|
| 509 |
+
row = {
|
| 510 |
+
"data_source": "timeseries_anol",
|
| 511 |
+
"prompt": [{"role": "user", "content": prompt_text}],
|
| 512 |
+
"images": [image],
|
| 513 |
+
"ability": "time_series_anomaly_detection",
|
| 514 |
+
"reward_model": {"style": "rule", "ground_truth": gt_intervals},
|
| 515 |
+
"extra_info": {
|
| 516 |
+
"category": task,
|
| 517 |
+
"split": split,
|
| 518 |
+
"instance_index": i,
|
| 519 |
+
"image_path": png_path,
|
| 520 |
+
"expcot": expcot,
|
| 521 |
+
"detection_metrics": detection_metrics,
|
| 522 |
+
"gt_intervals": gt_intervals,
|
| 523 |
+
"series_length": len(ts) if ts is not None else 0,
|
| 524 |
+
"index": pkl_i,
|
| 525 |
+
"anomaly_type": anomaly_type
|
| 526 |
+
}
|
| 527 |
+
}
|
| 528 |
+
results[split].append(row)
|
| 529 |
+
|
| 530 |
+
os.makedirs(out_dir, exist_ok=True)
|
| 531 |
+
train_path = os.path.join(out_dir, f"{task}_train.parquet")
|
| 532 |
+
test_path = os.path.join(out_dir, f"{task}_test.parquet")
|
| 533 |
+
|
| 534 |
+
if results["train"]:
|
| 535 |
+
train_ds = Dataset.from_list(results["train"])
|
| 536 |
+
train_ds.to_parquet(train_path)
|
| 537 |
+
else:
|
| 538 |
+
train_path = ""
|
| 539 |
+
|
| 540 |
+
if results["eval"]:
|
| 541 |
+
eval_ds = Dataset.from_list(results["eval"])
|
| 542 |
+
eval_ds.to_parquet(test_path)
|
| 543 |
+
else:
|
| 544 |
+
test_path = None
|
| 545 |
+
|
| 546 |
+
return train_path, test_path
|
| 547 |
+
|
| 548 |
+
|
| 549 |
+
|
| 550 |
+
def main():
|
| 551 |
+
parser = argparse.ArgumentParser(description="Process synthetic time series anomaly data into parquet files.")
|
| 552 |
+
parser.add_argument("--root_dir", type=str,
|
| 553 |
+
default="./data/anomllm/data/synthetic",
|
| 554 |
+
help="Root directory containing the task folders.")
|
| 555 |
+
parser.add_argument("--out_dir", type=str,
|
| 556 |
+
default="./data/anol_processed_mllm_data",
|
| 557 |
+
help="Output directory to store the final parquet files.")
|
| 558 |
+
args = parser.parse_args()
|
| 559 |
+
|
| 560 |
+
tasks = ["trend", "freq", "point", "range", "noisy-trend", "noisy-freq", "noisy-point", "flat-trend"]
|
| 561 |
+
|
| 562 |
+
print(f"Input data root: {args.root_dir}")
|
| 563 |
+
print(f"Output directory: {args.out_dir}")
|
| 564 |
+
os.makedirs(args.out_dir, exist_ok=True)
|
| 565 |
+
|
| 566 |
+
all_train_dfs = []
|
| 567 |
+
all_test_dfs = []
|
| 568 |
+
|
| 569 |
+
for task in tasks:
|
| 570 |
+
print(f"\nProcessing task: {task}...")
|
| 571 |
+
|
| 572 |
+
tr_path, te_path = process_task(args.root_dir, task, args.out_dir)
|
| 573 |
+
if tr_path and os.path.exists(tr_path):
|
| 574 |
+
print(f" ✅ Saved train data to -> {tr_path}")
|
| 575 |
+
all_train_dfs.append(pd.read_parquet(tr_path))
|
| 576 |
+
if te_path and os.path.exists(te_path):
|
| 577 |
+
print(f" ✅ Saved test data to -> {te_path}")
|
| 578 |
+
all_test_dfs.append(pd.read_parquet(te_path))
|
| 579 |
+
if not tr_path and not te_path:
|
| 580 |
+
print(f" ⚠️ No data found for task '{task}'. Check directory structure.")
|
| 581 |
+
|
| 582 |
+
|
| 583 |
+
if all_train_dfs:
|
| 584 |
+
full_train_df = pd.concat(all_train_dfs, ignore_index=True)
|
| 585 |
+
full_train_path = os.path.join(args.out_dir, "train_full.parquet")
|
| 586 |
+
full_train_df.to_parquet(full_train_path, index=False)
|
| 587 |
+
print(f"\n📦 Successfully merged and saved all training data to -> {full_train_path}")
|
| 588 |
+
|
| 589 |
+
if all_test_dfs:
|
| 590 |
+
full_test_df = pd.concat(all_test_dfs, ignore_index=True)
|
| 591 |
+
full_test_path = os.path.join(args.out_dir, "test_full.parquet")
|
| 592 |
+
full_test_df.to_parquet(full_test_path, index=False)
|
| 593 |
+
print(f"📦 Successfully merged and saved all testing data to -> {full_test_path}")
|
| 594 |
+
|
| 595 |
+
if __name__ == "__main__":
|
| 596 |
+
main()
|
multimodal_data_processing/rats_uni.py
ADDED
|
@@ -0,0 +1,268 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# -*- coding: utf-8 -*-
|
| 2 |
+
"""
|
| 3 |
+
Convert the Time-RA RATs-Uni (univariate) reasoning dataset into the parquet
|
| 4 |
+
format consumed by AnomSeer's veRL training/eval pipeline.
|
| 5 |
+
|
| 6 |
+
Unlike AnomLLM (see ``anom.py``), the Time-RA univariate dataset does **not**
|
| 7 |
+
provide anomaly *intervals* — it provides, per univariate series:
|
| 8 |
+
|
| 9 |
+
* ``Observation`` : the raw time series (list of floats, length 32/64/128)
|
| 10 |
+
* ``FigurePath`` : a rendered plot (``figures/{train,test}/{idx}.jpg``)
|
| 11 |
+
* ``ActionID`` : the anomaly class id (0-14, 0 == normal)
|
| 12 |
+
* ``Action`` : the human-readable class name
|
| 13 |
+
* ``Label`` : "Normal" / "Anomaly"
|
| 14 |
+
* ``Thought`` : an expert chain-of-thought explanation
|
| 15 |
+
|
| 16 |
+
We therefore adapt AnomSeer to a **classification + reasoning** task (no
|
| 17 |
+
localization). Each output row matches the schema expected by
|
| 18 |
+
``verl.utils.dataset.rl_dataset.RLHFDataset`` and is scored by the
|
| 19 |
+
``timeseries_rats`` reward (``verl/utils/reward_score/rats.py``):
|
| 20 |
+
|
| 21 |
+
data_source : "timeseries_rats"
|
| 22 |
+
prompt : [{"role": "user", "content": "<image>\\n ... <class> ..."}]
|
| 23 |
+
images : [{"bytes": <jpg bytes>, "path": <abs path>}]
|
| 24 |
+
ability : "time_series_anomaly_detection"
|
| 25 |
+
reward_model : {"style": "rule", "ground_truth": <canonical class name>}
|
| 26 |
+
extra_info : {category, source, split, anomaly_type, action_id, label,
|
| 27 |
+
series_length, image_path, expcot, numtext, index}
|
| 28 |
+
|
| 29 |
+
The ``numtext`` field is the expert CoT and is **required** by TimerPO's
|
| 30 |
+
semantic-alignment term (``compute_hidden_states_of_hint`` in fsdp_workers.py).
|
| 31 |
+
|
| 32 |
+
Usage
|
| 33 |
+
-----
|
| 34 |
+
python multimodal_data_processing/rats_uni.py \
|
| 35 |
+
--json_path /path/to/RATs40K/RATs-Uni-TSImage_Reason.json \
|
| 36 |
+
--out_dir ./data/rats_uni_processed
|
| 37 |
+
|
| 38 |
+
Produces ``train_full.parquet`` and ``test_full.parquet`` in ``--out_dir``.
|
| 39 |
+
"""
|
| 40 |
+
|
| 41 |
+
import os
|
| 42 |
+
import io
|
| 43 |
+
import json
|
| 44 |
+
import argparse
|
| 45 |
+
from typing import Dict, List, Optional, Tuple
|
| 46 |
+
|
| 47 |
+
import numpy as np
|
| 48 |
+
import pandas as pd
|
| 49 |
+
from PIL import Image
|
| 50 |
+
|
| 51 |
+
|
| 52 |
+
# Canonical 15-class taxonomy (id -> (name, description)), matching
|
| 53 |
+
# Time-RA/eval_utils.py::ANOMALY_DICT exactly.
|
| 54 |
+
ANOMALY_DICT: Dict[int, Tuple[str, str]] = {
|
| 55 |
+
0: ("Normal Sequence", "There are no abnormal situations in this time series."),
|
| 56 |
+
1: ("Point Anomaly", "A single data point significantly deviates from the local or global pattern of the sequence."),
|
| 57 |
+
2: ("Periodic Change Anomaly", "The original periodic pattern is disrupted, e.g. the period is broken or the amplitude becomes anomalous."),
|
| 58 |
+
3: ("Trend Change Anomaly", "A sudden change in the long-term trend of the time series."),
|
| 59 |
+
4: ("Change Point Anomaly", "Statistical properties (e.g. mean, variance) change abruptly at certain points."),
|
| 60 |
+
5: ("Distributional Change Anomaly", "The statistical distribution of the time series changes significantly."),
|
| 61 |
+
6: ("Amplitude Anomaly", "The amplitude of data points exceeds the normal upper and lower bounds."),
|
| 62 |
+
7: ("Pattern Change Anomaly", "The pattern of the time series suddenly changes from one form to another."),
|
| 63 |
+
8: ("Sparse Anomaly", "Isolated anomalous patterns occasionally appear in a long time series."),
|
| 64 |
+
9: ("Repeated Value Anomaly", "Continuous or intermittent repeated values disrupt the normal fluctuation pattern."),
|
| 65 |
+
10: ("Sudden Flatline Anomaly", "The time series suddenly becomes a flat line with no normal fluctuations."),
|
| 66 |
+
11: ("Drift Anomaly", "The data gradually drifts away from the normal level."),
|
| 67 |
+
12: ("Sudden Spike Anomaly", "The data suddenly spikes or drops within a short time and then returns to normal."),
|
| 68 |
+
13: ("Continuous Segment Anomaly", "A continuous segment of data points deviates from the normal pattern."),
|
| 69 |
+
14: ("Nonlinear Pattern Anomaly", "Nonlinear changes appear in the sequence, breaking the original linear rule."),
|
| 70 |
+
}
|
| 71 |
+
|
| 72 |
+
NAME_BY_ID = {i: name for i, (name, _) in ANOMALY_DICT.items()}
|
| 73 |
+
ID_BY_NAME = {name.lower(): i for i, (name, _) in ANOMALY_DICT.items()}
|
| 74 |
+
|
| 75 |
+
# Splits in the JSON -> output split name used by AnomSeer ("eval" == val/test).
|
| 76 |
+
SPLIT_MAP = {"TSAD_train": "train", "TSAD_test": "eval"}
|
| 77 |
+
|
| 78 |
+
|
| 79 |
+
def _build_taxonomy_block() -> str:
|
| 80 |
+
lines = []
|
| 81 |
+
for i in range(len(ANOMALY_DICT)):
|
| 82 |
+
name, desc = ANOMALY_DICT[i]
|
| 83 |
+
lines.append(f"{i}: {name} — {desc}")
|
| 84 |
+
return "\n".join(lines)
|
| 85 |
+
|
| 86 |
+
|
| 87 |
+
_TAXONOMY_BLOCK = _build_taxonomy_block()
|
| 88 |
+
_CLASS_NAMES = ", ".join(NAME_BY_ID[i] for i in range(len(NAME_BY_ID)))
|
| 89 |
+
|
| 90 |
+
|
| 91 |
+
def build_prompt(length: int, source: str) -> str:
|
| 92 |
+
"""Classification + reasoning prompt (no interval localization)."""
|
| 93 |
+
return (
|
| 94 |
+
"<image>\n"
|
| 95 |
+
f"You are an expert in univariate time-series anomaly detection. The figure shows a "
|
| 96 |
+
f"single-channel time series of length {length} from the \"{source}\" domain.\n\n"
|
| 97 |
+
"Decide whether the series is normal or contains an anomaly. If it is anomalous, choose the "
|
| 98 |
+
"single most appropriate anomaly type from the following 15 categories "
|
| 99 |
+
"(format `id: name — description`):\n"
|
| 100 |
+
f"{_TAXONOMY_BLOCK}\n\n"
|
| 101 |
+
"Reason step by step inside <think>...</think> based on the visual shape of the series, "
|
| 102 |
+
"then output exactly one line with your final answer:\n"
|
| 103 |
+
"<class>one exact category name from the list above</class>\n"
|
| 104 |
+
"If the series is normal, use <class>Normal Sequence</class>."
|
| 105 |
+
)
|
| 106 |
+
|
| 107 |
+
|
| 108 |
+
def _canonical_class(action: Optional[str], action_id: Optional[int]) -> Tuple[str, int]:
|
| 109 |
+
"""Resolve the canonical (name, id) from the raw Action / ActionID fields.
|
| 110 |
+
|
| 111 |
+
ActionID is treated as authoritative; Action is used only as a fallback.
|
| 112 |
+
"""
|
| 113 |
+
if isinstance(action_id, (int, np.integer)) and int(action_id) in NAME_BY_ID:
|
| 114 |
+
cid = int(action_id)
|
| 115 |
+
return NAME_BY_ID[cid], cid
|
| 116 |
+
if isinstance(action, str) and action.strip().lower() in ID_BY_NAME:
|
| 117 |
+
cid = ID_BY_NAME[action.strip().lower()]
|
| 118 |
+
return NAME_BY_ID[cid], cid
|
| 119 |
+
# Unknown -> treat as normal (id 0) so downstream typing stays consistent.
|
| 120 |
+
return NAME_BY_ID[0], 0
|
| 121 |
+
|
| 122 |
+
|
| 123 |
+
def _read_image_bytes(path: str) -> Optional[bytes]:
|
| 124 |
+
"""Return raw image bytes, re-encoding to a clean RGB JPEG if needed."""
|
| 125 |
+
if not os.path.isfile(path):
|
| 126 |
+
return None
|
| 127 |
+
try:
|
| 128 |
+
with open(path, "rb") as f:
|
| 129 |
+
raw = f.read()
|
| 130 |
+
# Validate; re-encode non-RGB to avoid downstream surprises.
|
| 131 |
+
with Image.open(io.BytesIO(raw)) as im:
|
| 132 |
+
if im.mode == "RGB":
|
| 133 |
+
return raw
|
| 134 |
+
buf = io.BytesIO()
|
| 135 |
+
im.convert("RGB").save(buf, format="JPEG", quality=95)
|
| 136 |
+
return buf.getvalue()
|
| 137 |
+
except Exception as exc: # noqa: BLE001
|
| 138 |
+
print(f"[WARN] failed to read image {path}: {exc}")
|
| 139 |
+
return None
|
| 140 |
+
|
| 141 |
+
|
| 142 |
+
# Some test entries point FigurePath at a ``.pdf`` that was never exported; the
|
| 143 |
+
# rendered raster lives next to it as a same-index ``.jpg``. Fall back across
|
| 144 |
+
# common raster extensions before giving up.
|
| 145 |
+
_IMG_EXT_FALLBACKS = (".jpg", ".png", ".jpeg")
|
| 146 |
+
|
| 147 |
+
|
| 148 |
+
def _resolve_figure_path(data_root: str, figure_path: str) -> str:
|
| 149 |
+
"""FigurePath is stored relative to the dataset root (e.g. figures/train/0.jpg).
|
| 150 |
+
|
| 151 |
+
Returns the first existing file, trying the stored path first and then the
|
| 152 |
+
same stem with a raster extension (handles ``.pdf`` FigurePaths)."""
|
| 153 |
+
base = figure_path if os.path.isabs(figure_path) else os.path.join(data_root, figure_path)
|
| 154 |
+
if os.path.isfile(base):
|
| 155 |
+
return base
|
| 156 |
+
stem, _ = os.path.splitext(base)
|
| 157 |
+
for ext in _IMG_EXT_FALLBACKS:
|
| 158 |
+
cand = stem + ext
|
| 159 |
+
if os.path.isfile(cand):
|
| 160 |
+
return cand
|
| 161 |
+
return base # non-existent; caller reports it as a skip
|
| 162 |
+
|
| 163 |
+
|
| 164 |
+
def process_split(records: dict, split_name: str, data_root: str,
|
| 165 |
+
max_samples: Optional[int] = None) -> List[dict]:
|
| 166 |
+
rows: List[dict] = []
|
| 167 |
+
n_skipped_none = 0
|
| 168 |
+
n_skipped_img = 0
|
| 169 |
+
|
| 170 |
+
# Iterate in numeric key order for reproducibility.
|
| 171 |
+
keys = sorted(records.keys(), key=lambda k: int(k) if str(k).isdigit() else k)
|
| 172 |
+
for out_idx, key in enumerate(keys):
|
| 173 |
+
if max_samples is not None and len(rows) >= max_samples:
|
| 174 |
+
break
|
| 175 |
+
|
| 176 |
+
entry = records[key]
|
| 177 |
+
if entry is None: # the JSON contains a handful of null test entries
|
| 178 |
+
n_skipped_none += 1
|
| 179 |
+
continue
|
| 180 |
+
|
| 181 |
+
obs = entry.get("Observation") or []
|
| 182 |
+
length = len(obs)
|
| 183 |
+
source = entry.get("Source", "unknown")
|
| 184 |
+
figure_path = entry.get("FigurePath", "")
|
| 185 |
+
thought = entry.get("Thought", "") or ""
|
| 186 |
+
label = entry.get("Label", "Anomaly")
|
| 187 |
+
|
| 188 |
+
img_path = _resolve_figure_path(data_root, figure_path)
|
| 189 |
+
img_bytes = _read_image_bytes(img_path)
|
| 190 |
+
if img_bytes is None:
|
| 191 |
+
n_skipped_img += 1
|
| 192 |
+
continue
|
| 193 |
+
|
| 194 |
+
class_name, class_id = _canonical_class(entry.get("Action"), entry.get("ActionID"))
|
| 195 |
+
prompt_text = build_prompt(length, source)
|
| 196 |
+
|
| 197 |
+
rows.append({
|
| 198 |
+
"data_source": "timeseries_rats",
|
| 199 |
+
"prompt": [{"role": "user", "content": prompt_text}],
|
| 200 |
+
"images": [{"bytes": img_bytes, "path": img_path}],
|
| 201 |
+
"ability": "time_series_anomaly_detection",
|
| 202 |
+
"reward_model": {"style": "rule", "ground_truth": class_name},
|
| 203 |
+
"extra_info": {
|
| 204 |
+
"index": int(key) if str(key).isdigit() else out_idx,
|
| 205 |
+
"category": class_name, # per-class metric grouping
|
| 206 |
+
"source": source, # domain grouping
|
| 207 |
+
"split": split_name,
|
| 208 |
+
"anomaly_type": class_name, # read by the reward as GT class
|
| 209 |
+
"action_id": int(class_id),
|
| 210 |
+
"label": label,
|
| 211 |
+
"series_length": int(length),
|
| 212 |
+
"image_path": img_path,
|
| 213 |
+
"expcot": thought, # expert CoT
|
| 214 |
+
"numtext": thought, # required by TimerPO hint term
|
| 215 |
+
},
|
| 216 |
+
})
|
| 217 |
+
|
| 218 |
+
print(f" [{split_name}] kept={len(rows)} "
|
| 219 |
+
f"skipped(null)={n_skipped_none} skipped(missing-image)={n_skipped_img}")
|
| 220 |
+
return rows
|
| 221 |
+
|
| 222 |
+
|
| 223 |
+
def write_parquet(rows: List[dict], path: str) -> None:
|
| 224 |
+
if not rows:
|
| 225 |
+
print(f" [skip] no rows for {path}")
|
| 226 |
+
return
|
| 227 |
+
df = pd.DataFrame(rows)
|
| 228 |
+
df.to_parquet(path, index=False, engine="pyarrow")
|
| 229 |
+
print(f" ✅ wrote {len(df)} rows -> {path}")
|
| 230 |
+
|
| 231 |
+
|
| 232 |
+
def main() -> None:
|
| 233 |
+
parser = argparse.ArgumentParser(
|
| 234 |
+
description="Convert Time-RA RATs-Uni JSON into AnomSeer parquet (classification + reasoning).")
|
| 235 |
+
parser.add_argument("--json_path", type=str,
|
| 236 |
+
default="/mnt/share01/sqk/datasets/RATs40K/RATs-Uni-TSImage_Reason.json",
|
| 237 |
+
help="Path to RATs-Uni-TSImage_Reason.json")
|
| 238 |
+
parser.add_argument("--data_root", type=str, default=None,
|
| 239 |
+
help="Dataset root that FigurePath is relative to "
|
| 240 |
+
"(defaults to the directory of --json_path).")
|
| 241 |
+
parser.add_argument("--out_dir", type=str, default="./data/rats_uni_processed",
|
| 242 |
+
help="Output directory for the parquet files.")
|
| 243 |
+
parser.add_argument("--max_samples", type=int, default=None,
|
| 244 |
+
help="Optional cap on samples per split (for quick tests).")
|
| 245 |
+
args = parser.parse_args()
|
| 246 |
+
|
| 247 |
+
data_root = args.data_root or os.path.dirname(os.path.abspath(args.json_path))
|
| 248 |
+
os.makedirs(args.out_dir, exist_ok=True)
|
| 249 |
+
|
| 250 |
+
print(f"Reading {args.json_path}")
|
| 251 |
+
print(f"Figure root: {data_root}")
|
| 252 |
+
with open(args.json_path, "r") as f:
|
| 253 |
+
data = json.load(f)
|
| 254 |
+
|
| 255 |
+
for json_split, out_split in SPLIT_MAP.items():
|
| 256 |
+
if json_split not in data:
|
| 257 |
+
print(f"[WARN] split '{json_split}' not found in JSON; skipping.")
|
| 258 |
+
continue
|
| 259 |
+
print(f"\nProcessing {json_split} -> {out_split}")
|
| 260 |
+
rows = process_split(data[json_split], out_split, data_root, args.max_samples)
|
| 261 |
+
fname = "train_full.parquet" if out_split == "train" else "test_full.parquet"
|
| 262 |
+
write_parquet(rows, os.path.join(args.out_dir, fname))
|
| 263 |
+
|
| 264 |
+
print("\nDone.")
|
| 265 |
+
|
| 266 |
+
|
| 267 |
+
if __name__ == "__main__":
|
| 268 |
+
main()
|
outputs/2026-06-09/15-29-24/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_full.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: true
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: all-linear
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: true
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/15-29-24/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules=all-linear
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=True
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 148 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 149 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 150 |
+
- trainer.critic_warmup=0
|
| 151 |
+
- trainer.logger=['console']
|
| 152 |
+
- trainer.project_name=anomseer
|
| 153 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 154 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.n_gpus_per_node=2
|
| 156 |
+
- trainer.nnodes=1
|
| 157 |
+
- trainer.save_freq=500
|
| 158 |
+
- trainer.test_freq=10
|
| 159 |
+
- trainer.val_only=False
|
| 160 |
+
- trainer.val_before_train=True
|
| 161 |
+
- trainer.total_epochs=10
|
| 162 |
+
- ts.use_sem_orth=True
|
| 163 |
+
- ts.adv_mix=0.3
|
| 164 |
+
- ts.similarity_method=ot
|
| 165 |
+
- ts.ot_eps=0.08
|
| 166 |
+
- ts.ot_n_iter=50
|
| 167 |
+
job:
|
| 168 |
+
name: main_ppo
|
| 169 |
+
chdir: null
|
| 170 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules=all-linear,actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct,actor_rollout_ref.model.use_remove_padding=True,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_full.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 171 |
+
id: ???
|
| 172 |
+
num: ???
|
| 173 |
+
config_name: ppo_trainer
|
| 174 |
+
env_set: {}
|
| 175 |
+
env_copy: []
|
| 176 |
+
config:
|
| 177 |
+
override_dirname:
|
| 178 |
+
kv_sep: '='
|
| 179 |
+
item_sep: ','
|
| 180 |
+
exclude_keys: []
|
| 181 |
+
runtime:
|
| 182 |
+
version: 1.3.0
|
| 183 |
+
version_base: '1.3'
|
| 184 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 185 |
+
config_sources:
|
| 186 |
+
- path: hydra.conf
|
| 187 |
+
schema: pkg
|
| 188 |
+
provider: hydra
|
| 189 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 190 |
+
schema: file
|
| 191 |
+
provider: main
|
| 192 |
+
- path: ''
|
| 193 |
+
schema: structured
|
| 194 |
+
provider: schema
|
| 195 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/15-29-24
|
| 196 |
+
choices:
|
| 197 |
+
hydra/env: default
|
| 198 |
+
hydra/callbacks: null
|
| 199 |
+
hydra/job_logging: default
|
| 200 |
+
hydra/hydra_logging: default
|
| 201 |
+
hydra/hydra_help: default
|
| 202 |
+
hydra/help: default
|
| 203 |
+
hydra/sweeper: basic
|
| 204 |
+
hydra/launcher: basic
|
| 205 |
+
hydra/output: default
|
| 206 |
+
verbose: false
|
outputs/2026-06-09/15-29-24/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules=all-linear
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=True
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 34 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 35 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 36 |
+
- trainer.critic_warmup=0
|
| 37 |
+
- trainer.logger=['console']
|
| 38 |
+
- trainer.project_name=anomseer
|
| 39 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 40 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.n_gpus_per_node=2
|
| 42 |
+
- trainer.nnodes=1
|
| 43 |
+
- trainer.save_freq=500
|
| 44 |
+
- trainer.test_freq=10
|
| 45 |
+
- trainer.val_only=False
|
| 46 |
+
- trainer.val_before_train=True
|
| 47 |
+
- trainer.total_epochs=10
|
| 48 |
+
- ts.use_sem_orth=True
|
| 49 |
+
- ts.adv_mix=0.3
|
| 50 |
+
- ts.similarity_method=ot
|
| 51 |
+
- ts.ot_eps=0.08
|
| 52 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/15-29-24/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/16-19-04/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_full.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: all-linear
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: true
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/16-19-04/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules=all-linear
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 148 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 149 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 150 |
+
- trainer.critic_warmup=0
|
| 151 |
+
- trainer.logger=['console']
|
| 152 |
+
- trainer.project_name=anomseer
|
| 153 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 154 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.n_gpus_per_node=2
|
| 156 |
+
- trainer.nnodes=1
|
| 157 |
+
- trainer.save_freq=500
|
| 158 |
+
- trainer.test_freq=10
|
| 159 |
+
- trainer.val_only=False
|
| 160 |
+
- trainer.val_before_train=True
|
| 161 |
+
- trainer.total_epochs=10
|
| 162 |
+
- ts.use_sem_orth=True
|
| 163 |
+
- ts.adv_mix=0.3
|
| 164 |
+
- ts.similarity_method=ot
|
| 165 |
+
- ts.ot_eps=0.08
|
| 166 |
+
- ts.ot_n_iter=50
|
| 167 |
+
job:
|
| 168 |
+
name: main_ppo
|
| 169 |
+
chdir: null
|
| 170 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules=all-linear,actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_full.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 171 |
+
id: ???
|
| 172 |
+
num: ???
|
| 173 |
+
config_name: ppo_trainer
|
| 174 |
+
env_set: {}
|
| 175 |
+
env_copy: []
|
| 176 |
+
config:
|
| 177 |
+
override_dirname:
|
| 178 |
+
kv_sep: '='
|
| 179 |
+
item_sep: ','
|
| 180 |
+
exclude_keys: []
|
| 181 |
+
runtime:
|
| 182 |
+
version: 1.3.0
|
| 183 |
+
version_base: '1.3'
|
| 184 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 185 |
+
config_sources:
|
| 186 |
+
- path: hydra.conf
|
| 187 |
+
schema: pkg
|
| 188 |
+
provider: hydra
|
| 189 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 190 |
+
schema: file
|
| 191 |
+
provider: main
|
| 192 |
+
- path: ''
|
| 193 |
+
schema: structured
|
| 194 |
+
provider: schema
|
| 195 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/16-19-04
|
| 196 |
+
choices:
|
| 197 |
+
hydra/env: default
|
| 198 |
+
hydra/callbacks: null
|
| 199 |
+
hydra/job_logging: default
|
| 200 |
+
hydra/hydra_logging: default
|
| 201 |
+
hydra/hydra_help: default
|
| 202 |
+
hydra/help: default
|
| 203 |
+
hydra/sweeper: basic
|
| 204 |
+
hydra/launcher: basic
|
| 205 |
+
hydra/output: default
|
| 206 |
+
verbose: false
|
outputs/2026-06-09/16-19-04/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules=all-linear
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 34 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 35 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 36 |
+
- trainer.critic_warmup=0
|
| 37 |
+
- trainer.logger=['console']
|
| 38 |
+
- trainer.project_name=anomseer
|
| 39 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 40 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.n_gpus_per_node=2
|
| 42 |
+
- trainer.nnodes=1
|
| 43 |
+
- trainer.save_freq=500
|
| 44 |
+
- trainer.test_freq=10
|
| 45 |
+
- trainer.val_only=False
|
| 46 |
+
- trainer.val_before_train=True
|
| 47 |
+
- trainer.total_epochs=10
|
| 48 |
+
- ts.use_sem_orth=True
|
| 49 |
+
- ts.adv_mix=0.3
|
| 50 |
+
- ts.similarity_method=ot
|
| 51 |
+
- ts.ot_eps=0.08
|
| 52 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/16-19-04/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/16-56-49/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_full.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: true
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/16-56-49/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 148 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 149 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 150 |
+
- trainer.critic_warmup=0
|
| 151 |
+
- trainer.logger=['console']
|
| 152 |
+
- trainer.project_name=anomseer
|
| 153 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 154 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.n_gpus_per_node=2
|
| 156 |
+
- trainer.nnodes=1
|
| 157 |
+
- trainer.save_freq=500
|
| 158 |
+
- trainer.test_freq=10
|
| 159 |
+
- trainer.val_only=False
|
| 160 |
+
- trainer.val_before_train=True
|
| 161 |
+
- trainer.total_epochs=10
|
| 162 |
+
- ts.use_sem_orth=True
|
| 163 |
+
- ts.adv_mix=0.3
|
| 164 |
+
- ts.similarity_method=ot
|
| 165 |
+
- ts.ot_eps=0.08
|
| 166 |
+
- ts.ot_n_iter=50
|
| 167 |
+
job:
|
| 168 |
+
name: main_ppo
|
| 169 |
+
chdir: null
|
| 170 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_full.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 171 |
+
id: ???
|
| 172 |
+
num: ???
|
| 173 |
+
config_name: ppo_trainer
|
| 174 |
+
env_set: {}
|
| 175 |
+
env_copy: []
|
| 176 |
+
config:
|
| 177 |
+
override_dirname:
|
| 178 |
+
kv_sep: '='
|
| 179 |
+
item_sep: ','
|
| 180 |
+
exclude_keys: []
|
| 181 |
+
runtime:
|
| 182 |
+
version: 1.3.0
|
| 183 |
+
version_base: '1.3'
|
| 184 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 185 |
+
config_sources:
|
| 186 |
+
- path: hydra.conf
|
| 187 |
+
schema: pkg
|
| 188 |
+
provider: hydra
|
| 189 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 190 |
+
schema: file
|
| 191 |
+
provider: main
|
| 192 |
+
- path: ''
|
| 193 |
+
schema: structured
|
| 194 |
+
provider: schema
|
| 195 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/16-56-49
|
| 196 |
+
choices:
|
| 197 |
+
hydra/env: default
|
| 198 |
+
hydra/callbacks: null
|
| 199 |
+
hydra/job_logging: default
|
| 200 |
+
hydra/hydra_logging: default
|
| 201 |
+
hydra/hydra_help: default
|
| 202 |
+
hydra/help: default
|
| 203 |
+
hydra/sweeper: basic
|
| 204 |
+
hydra/launcher: basic
|
| 205 |
+
hydra/output: default
|
| 206 |
+
verbose: false
|
outputs/2026-06-09/16-56-49/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 34 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 35 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 36 |
+
- trainer.critic_warmup=0
|
| 37 |
+
- trainer.logger=['console']
|
| 38 |
+
- trainer.project_name=anomseer
|
| 39 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 40 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.n_gpus_per_node=2
|
| 42 |
+
- trainer.nnodes=1
|
| 43 |
+
- trainer.save_freq=500
|
| 44 |
+
- trainer.test_freq=10
|
| 45 |
+
- trainer.val_only=False
|
| 46 |
+
- trainer.val_before_train=True
|
| 47 |
+
- trainer.total_epochs=10
|
| 48 |
+
- ts.use_sem_orth=True
|
| 49 |
+
- ts.adv_mix=0.3
|
| 50 |
+
- ts.similarity_method=ot
|
| 51 |
+
- ts.ot_eps=0.08
|
| 52 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/16-56-49/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/17-08-35/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_full.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: true
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/17-08-35/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 148 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 149 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 150 |
+
- trainer.critic_warmup=0
|
| 151 |
+
- trainer.logger=['console']
|
| 152 |
+
- trainer.project_name=anomseer
|
| 153 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 154 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.n_gpus_per_node=2
|
| 156 |
+
- trainer.nnodes=1
|
| 157 |
+
- trainer.save_freq=500
|
| 158 |
+
- trainer.test_freq=10
|
| 159 |
+
- trainer.val_only=False
|
| 160 |
+
- trainer.val_before_train=True
|
| 161 |
+
- trainer.total_epochs=10
|
| 162 |
+
- ts.use_sem_orth=True
|
| 163 |
+
- ts.adv_mix=0.3
|
| 164 |
+
- ts.similarity_method=ot
|
| 165 |
+
- ts.ot_eps=0.08
|
| 166 |
+
- ts.ot_n_iter=50
|
| 167 |
+
job:
|
| 168 |
+
name: main_ppo
|
| 169 |
+
chdir: null
|
| 170 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_full.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 171 |
+
id: ???
|
| 172 |
+
num: ???
|
| 173 |
+
config_name: ppo_trainer
|
| 174 |
+
env_set: {}
|
| 175 |
+
env_copy: []
|
| 176 |
+
config:
|
| 177 |
+
override_dirname:
|
| 178 |
+
kv_sep: '='
|
| 179 |
+
item_sep: ','
|
| 180 |
+
exclude_keys: []
|
| 181 |
+
runtime:
|
| 182 |
+
version: 1.3.0
|
| 183 |
+
version_base: '1.3'
|
| 184 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 185 |
+
config_sources:
|
| 186 |
+
- path: hydra.conf
|
| 187 |
+
schema: pkg
|
| 188 |
+
provider: hydra
|
| 189 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 190 |
+
schema: file
|
| 191 |
+
provider: main
|
| 192 |
+
- path: ''
|
| 193 |
+
schema: structured
|
| 194 |
+
provider: schema
|
| 195 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/17-08-35
|
| 196 |
+
choices:
|
| 197 |
+
hydra/env: default
|
| 198 |
+
hydra/callbacks: null
|
| 199 |
+
hydra/job_logging: default
|
| 200 |
+
hydra/hydra_logging: default
|
| 201 |
+
hydra/hydra_help: default
|
| 202 |
+
hydra/help: default
|
| 203 |
+
hydra/sweeper: basic
|
| 204 |
+
hydra/launcher: basic
|
| 205 |
+
hydra/output: default
|
| 206 |
+
verbose: false
|
outputs/2026-06-09/17-08-35/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/qwen2.5-vl-7b-instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 34 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 35 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 36 |
+
- trainer.critic_warmup=0
|
| 37 |
+
- trainer.logger=['console']
|
| 38 |
+
- trainer.project_name=anomseer
|
| 39 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 40 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.n_gpus_per_node=2
|
| 42 |
+
- trainer.nnodes=1
|
| 43 |
+
- trainer.save_freq=500
|
| 44 |
+
- trainer.test_freq=10
|
| 45 |
+
- trainer.val_only=False
|
| 46 |
+
- trainer.val_before_train=True
|
| 47 |
+
- trainer.total_epochs=10
|
| 48 |
+
- ts.use_sem_orth=True
|
| 49 |
+
- ts.adv_mix=0.3
|
| 50 |
+
- ts.similarity_method=ot
|
| 51 |
+
- ts.ot_eps=0.08
|
| 52 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/17-08-35/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/17-25-05/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_full.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: true
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/17-25-05/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 148 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 149 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 150 |
+
- trainer.critic_warmup=0
|
| 151 |
+
- trainer.logger=['console']
|
| 152 |
+
- trainer.project_name=anomseer
|
| 153 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 154 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.n_gpus_per_node=2
|
| 156 |
+
- trainer.nnodes=1
|
| 157 |
+
- trainer.save_freq=500
|
| 158 |
+
- trainer.test_freq=10
|
| 159 |
+
- trainer.val_only=False
|
| 160 |
+
- trainer.val_before_train=True
|
| 161 |
+
- trainer.total_epochs=10
|
| 162 |
+
- ts.use_sem_orth=True
|
| 163 |
+
- ts.adv_mix=0.3
|
| 164 |
+
- ts.similarity_method=ot
|
| 165 |
+
- ts.ot_eps=0.08
|
| 166 |
+
- ts.ot_n_iter=50
|
| 167 |
+
job:
|
| 168 |
+
name: main_ppo
|
| 169 |
+
chdir: null
|
| 170 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_full.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 171 |
+
id: ???
|
| 172 |
+
num: ???
|
| 173 |
+
config_name: ppo_trainer
|
| 174 |
+
env_set: {}
|
| 175 |
+
env_copy: []
|
| 176 |
+
config:
|
| 177 |
+
override_dirname:
|
| 178 |
+
kv_sep: '='
|
| 179 |
+
item_sep: ','
|
| 180 |
+
exclude_keys: []
|
| 181 |
+
runtime:
|
| 182 |
+
version: 1.3.0
|
| 183 |
+
version_base: '1.3'
|
| 184 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 185 |
+
config_sources:
|
| 186 |
+
- path: hydra.conf
|
| 187 |
+
schema: pkg
|
| 188 |
+
provider: hydra
|
| 189 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 190 |
+
schema: file
|
| 191 |
+
provider: main
|
| 192 |
+
- path: ''
|
| 193 |
+
schema: structured
|
| 194 |
+
provider: schema
|
| 195 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/17-25-05
|
| 196 |
+
choices:
|
| 197 |
+
hydra/env: default
|
| 198 |
+
hydra/callbacks: null
|
| 199 |
+
hydra/job_logging: default
|
| 200 |
+
hydra/hydra_logging: default
|
| 201 |
+
hydra/hydra_help: default
|
| 202 |
+
hydra/help: default
|
| 203 |
+
hydra/sweeper: basic
|
| 204 |
+
hydra/launcher: basic
|
| 205 |
+
hydra/output: default
|
| 206 |
+
verbose: false
|
outputs/2026-06-09/17-25-05/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_full.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 34 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 35 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 36 |
+
- trainer.critic_warmup=0
|
| 37 |
+
- trainer.logger=['console']
|
| 38 |
+
- trainer.project_name=anomseer
|
| 39 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 40 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.n_gpus_per_node=2
|
| 42 |
+
- trainer.nnodes=1
|
| 43 |
+
- trainer.save_freq=500
|
| 44 |
+
- trainer.test_freq=10
|
| 45 |
+
- trainer.val_only=False
|
| 46 |
+
- trainer.val_before_train=True
|
| 47 |
+
- trainer.total_epochs=10
|
| 48 |
+
- ts.use_sem_orth=True
|
| 49 |
+
- ts.adv_mix=0.3
|
| 50 |
+
- ts.similarity_method=ot
|
| 51 |
+
- ts.ot_eps=0.08
|
| 52 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/17-25-05/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/17-44-32/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_small.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: false
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/17-44-32/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,207 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=./data/rats_uni_processed/test_small.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 148 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 149 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 150 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 151 |
+
- trainer.critic_warmup=0
|
| 152 |
+
- trainer.logger=['console']
|
| 153 |
+
- trainer.project_name=anomseer
|
| 154 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 156 |
+
- trainer.n_gpus_per_node=2
|
| 157 |
+
- trainer.nnodes=1
|
| 158 |
+
- trainer.save_freq=500
|
| 159 |
+
- trainer.test_freq=10
|
| 160 |
+
- trainer.val_only=False
|
| 161 |
+
- trainer.val_before_train=True
|
| 162 |
+
- trainer.total_epochs=10
|
| 163 |
+
- ts.use_sem_orth=True
|
| 164 |
+
- ts.adv_mix=0.3
|
| 165 |
+
- ts.similarity_method=ot
|
| 166 |
+
- ts.ot_eps=0.08
|
| 167 |
+
- ts.ot_n_iter=50
|
| 168 |
+
job:
|
| 169 |
+
name: main_ppo
|
| 170 |
+
chdir: null
|
| 171 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,actor_rollout_ref.rollout.val_kwargs.do_sample=False,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=./data/rats_uni_processed/test_small.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 172 |
+
id: ???
|
| 173 |
+
num: ???
|
| 174 |
+
config_name: ppo_trainer
|
| 175 |
+
env_set: {}
|
| 176 |
+
env_copy: []
|
| 177 |
+
config:
|
| 178 |
+
override_dirname:
|
| 179 |
+
kv_sep: '='
|
| 180 |
+
item_sep: ','
|
| 181 |
+
exclude_keys: []
|
| 182 |
+
runtime:
|
| 183 |
+
version: 1.3.0
|
| 184 |
+
version_base: '1.3'
|
| 185 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 186 |
+
config_sources:
|
| 187 |
+
- path: hydra.conf
|
| 188 |
+
schema: pkg
|
| 189 |
+
provider: hydra
|
| 190 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 191 |
+
schema: file
|
| 192 |
+
provider: main
|
| 193 |
+
- path: ''
|
| 194 |
+
schema: structured
|
| 195 |
+
provider: schema
|
| 196 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/17-44-32
|
| 197 |
+
choices:
|
| 198 |
+
hydra/env: default
|
| 199 |
+
hydra/callbacks: null
|
| 200 |
+
hydra/job_logging: default
|
| 201 |
+
hydra/hydra_logging: default
|
| 202 |
+
hydra/hydra_help: default
|
| 203 |
+
hydra/help: default
|
| 204 |
+
hydra/sweeper: basic
|
| 205 |
+
hydra/launcher: basic
|
| 206 |
+
hydra/output: default
|
| 207 |
+
verbose: false
|
outputs/2026-06-09/17-44-32/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_small.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 34 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 35 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 36 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 37 |
+
- trainer.critic_warmup=0
|
| 38 |
+
- trainer.logger=['console']
|
| 39 |
+
- trainer.project_name=anomseer
|
| 40 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 42 |
+
- trainer.n_gpus_per_node=2
|
| 43 |
+
- trainer.nnodes=1
|
| 44 |
+
- trainer.save_freq=500
|
| 45 |
+
- trainer.test_freq=10
|
| 46 |
+
- trainer.val_only=False
|
| 47 |
+
- trainer.val_before_train=True
|
| 48 |
+
- trainer.total_epochs=10
|
| 49 |
+
- ts.use_sem_orth=True
|
| 50 |
+
- ts.adv_mix=0.3
|
| 51 |
+
- ts.similarity_method=ot
|
| 52 |
+
- ts.ot_eps=0.08
|
| 53 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/17-44-32/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-09/18-11-50/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: /tmp/anomseer_rats_one.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 16
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: 16
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 1
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: false
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_lora_smoke
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_lora_smoke
|
| 185 |
+
val_only: true
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/18-11-50/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,210 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 117 |
+
- data.val_files=/tmp/anomseer_rats_one.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 148 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 149 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 150 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 151 |
+
- trainer.critic_warmup=0
|
| 152 |
+
- trainer.logger=['console']
|
| 153 |
+
- trainer.project_name=anomseer
|
| 154 |
+
- trainer.experiment_name=anomseer_lora_smoke
|
| 155 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_lora_smoke
|
| 156 |
+
- trainer.n_gpus_per_node=2
|
| 157 |
+
- trainer.nnodes=1
|
| 158 |
+
- trainer.save_freq=500
|
| 159 |
+
- trainer.test_freq=10
|
| 160 |
+
- trainer.val_only=True
|
| 161 |
+
- trainer.val_before_train=True
|
| 162 |
+
- trainer.total_epochs=10
|
| 163 |
+
- ts.use_sem_orth=True
|
| 164 |
+
- ts.adv_mix=0.3
|
| 165 |
+
- ts.similarity_method=ot
|
| 166 |
+
- ts.ot_eps=0.08
|
| 167 |
+
- ts.ot_n_iter=50
|
| 168 |
+
- data.max_response_length=16
|
| 169 |
+
- actor_rollout_ref.rollout.response_length=16
|
| 170 |
+
- actor_rollout_ref.rollout.n=1
|
| 171 |
+
job:
|
| 172 |
+
name: main_ppo
|
| 173 |
+
chdir: null
|
| 174 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=1,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.response_length=16,actor_rollout_ref.rollout.tensor_model_parallel_size=2,actor_rollout_ref.rollout.val_kwargs.do_sample=False,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.max_response_length=16,data.train_batch_size=128,data.train_files=./data/rats_uni_processed/train_full.parquet,data.truncation=error,data.val_files=/tmp/anomseer_rats_one.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_lora_smoke,trainer.experiment_name=anomseer_lora_smoke,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=500,trainer.test_freq=10,trainer.total_epochs=10,trainer.val_before_train=True,trainer.val_only=True,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 175 |
+
id: ???
|
| 176 |
+
num: ???
|
| 177 |
+
config_name: ppo_trainer
|
| 178 |
+
env_set: {}
|
| 179 |
+
env_copy: []
|
| 180 |
+
config:
|
| 181 |
+
override_dirname:
|
| 182 |
+
kv_sep: '='
|
| 183 |
+
item_sep: ','
|
| 184 |
+
exclude_keys: []
|
| 185 |
+
runtime:
|
| 186 |
+
version: 1.3.0
|
| 187 |
+
version_base: '1.3'
|
| 188 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 189 |
+
config_sources:
|
| 190 |
+
- path: hydra.conf
|
| 191 |
+
schema: pkg
|
| 192 |
+
provider: hydra
|
| 193 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 194 |
+
schema: file
|
| 195 |
+
provider: main
|
| 196 |
+
- path: ''
|
| 197 |
+
schema: structured
|
| 198 |
+
provider: schema
|
| 199 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/18-11-50
|
| 200 |
+
choices:
|
| 201 |
+
hydra/env: default
|
| 202 |
+
hydra/callbacks: null
|
| 203 |
+
hydra/job_logging: default
|
| 204 |
+
hydra/hydra_logging: default
|
| 205 |
+
hydra/hydra_help: default
|
| 206 |
+
hydra/help: default
|
| 207 |
+
hydra/sweeper: basic
|
| 208 |
+
hydra/launcher: basic
|
| 209 |
+
hydra/output: default
|
| 210 |
+
verbose: false
|
outputs/2026-06-09/18-11-50/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=/tmp/anomseer_rats_one.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 34 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 35 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 36 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 37 |
+
- trainer.critic_warmup=0
|
| 38 |
+
- trainer.logger=['console']
|
| 39 |
+
- trainer.project_name=anomseer
|
| 40 |
+
- trainer.experiment_name=anomseer_lora_smoke
|
| 41 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_lora_smoke
|
| 42 |
+
- trainer.n_gpus_per_node=2
|
| 43 |
+
- trainer.nnodes=1
|
| 44 |
+
- trainer.save_freq=500
|
| 45 |
+
- trainer.test_freq=10
|
| 46 |
+
- trainer.val_only=True
|
| 47 |
+
- trainer.val_before_train=True
|
| 48 |
+
- trainer.total_epochs=10
|
| 49 |
+
- ts.use_sem_orth=True
|
| 50 |
+
- ts.adv_mix=0.3
|
| 51 |
+
- ts.similarity_method=ot
|
| 52 |
+
- ts.ot_eps=0.08
|
| 53 |
+
- ts.ot_n_iter=50
|
| 54 |
+
- data.max_response_length=16
|
| 55 |
+
- actor_rollout_ref.rollout.response_length=16
|
| 56 |
+
- actor_rollout_ref.rollout.n=1
|
outputs/2026-06-09/18-20-25/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: ./data/rats_uni_processed/train_full.parquet
|
| 4 |
+
val_files: ./data/rats_uni_processed/test_small.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 128
|
| 9 |
+
val_batch_size: null
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 16
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 128
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: false
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 10
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: auto
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: 10
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: false
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-09/18-20-25/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=./data/rats_uni_processed/train_full.parquet
|
| 3 |
+
- data.val_files=./data/rats_uni_processed/test_small.parquet
|
| 4 |
+
- data.train_batch_size=128
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 34 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 35 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 36 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 37 |
+
- trainer.critic_warmup=0
|
| 38 |
+
- trainer.logger=['console']
|
| 39 |
+
- trainer.project_name=anomseer
|
| 40 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 42 |
+
- trainer.n_gpus_per_node=2
|
| 43 |
+
- trainer.nnodes=1
|
| 44 |
+
- trainer.save_freq=500
|
| 45 |
+
- trainer.test_freq=10
|
| 46 |
+
- trainer.val_only=False
|
| 47 |
+
- trainer.val_before_train=True
|
| 48 |
+
- trainer.total_epochs=10
|
| 49 |
+
- ts.use_sem_orth=True
|
| 50 |
+
- ts.adv_mix=0.3
|
| 51 |
+
- ts.similarity_method=ot
|
| 52 |
+
- ts.ot_eps=0.08
|
| 53 |
+
- ts.ot_n_iter=50
|
outputs/2026-06-09/18-33-56/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,216 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=/tmp/anomseer_rats_train_one.parquet
|
| 117 |
+
- data.val_files=/tmp/anomseer_rats_val_one.parquet
|
| 118 |
+
- data.train_batch_size=128
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=16
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=128
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=1
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=1
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 148 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=1
|
| 149 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 150 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 151 |
+
- trainer.critic_warmup=0
|
| 152 |
+
- trainer.logger=['console']
|
| 153 |
+
- trainer.project_name=anomseer
|
| 154 |
+
- trainer.experiment_name=anomseer_lora_train_smoke
|
| 155 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_lora_train_smoke
|
| 156 |
+
- trainer.n_gpus_per_node=2
|
| 157 |
+
- trainer.nnodes=1
|
| 158 |
+
- trainer.save_freq=500
|
| 159 |
+
- trainer.test_freq=10
|
| 160 |
+
- trainer.val_only=False
|
| 161 |
+
- trainer.val_before_train=True
|
| 162 |
+
- trainer.total_epochs=1
|
| 163 |
+
- ts.use_sem_orth=True
|
| 164 |
+
- ts.adv_mix=0.3
|
| 165 |
+
- ts.similarity_method=ot
|
| 166 |
+
- ts.ot_eps=0.08
|
| 167 |
+
- ts.ot_n_iter=50
|
| 168 |
+
- data.train_batch_size=1
|
| 169 |
+
- data.max_response_length=16
|
| 170 |
+
- actor_rollout_ref.rollout.response_length=16
|
| 171 |
+
- actor_rollout_ref.rollout.n=1
|
| 172 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=1
|
| 173 |
+
- trainer.val_before_train=False
|
| 174 |
+
- trainer.total_training_steps=1
|
| 175 |
+
- trainer.save_freq=-1
|
| 176 |
+
- trainer.test_freq=-1
|
| 177 |
+
job:
|
| 178 |
+
name: main_ppo
|
| 179 |
+
chdir: null
|
| 180 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=1,actor_rollout_ref.actor.ppo_mini_batch_size=1,actor_rollout_ref.actor.ppo_mini_batch_size=128,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=16,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/models/Qwen2.5-VL-3B-Instruct,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=1,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=1,actor_rollout_ref.rollout.n=1,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.response_length=16,actor_rollout_ref.rollout.tensor_model_parallel_size=2,actor_rollout_ref.rollout.val_kwargs.do_sample=False,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.max_response_length=16,data.train_batch_size=1,data.train_batch_size=128,data.train_files=/tmp/anomseer_rats_train_one.parquet,data.truncation=error,data.val_files=/tmp/anomseer_rats_val_one.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_lora_train_smoke,trainer.experiment_name=anomseer_lora_train_smoke,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.save_freq=-1,trainer.save_freq=500,trainer.test_freq=-1,trainer.test_freq=10,trainer.total_epochs=1,trainer.total_training_steps=1,trainer.val_before_train=False,trainer.val_before_train=True,trainer.val_only=False,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 181 |
+
id: ???
|
| 182 |
+
num: ???
|
| 183 |
+
config_name: ppo_trainer
|
| 184 |
+
env_set: {}
|
| 185 |
+
env_copy: []
|
| 186 |
+
config:
|
| 187 |
+
override_dirname:
|
| 188 |
+
kv_sep: '='
|
| 189 |
+
item_sep: ','
|
| 190 |
+
exclude_keys: []
|
| 191 |
+
runtime:
|
| 192 |
+
version: 1.3.0
|
| 193 |
+
version_base: '1.3'
|
| 194 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 195 |
+
config_sources:
|
| 196 |
+
- path: hydra.conf
|
| 197 |
+
schema: pkg
|
| 198 |
+
provider: hydra
|
| 199 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 200 |
+
schema: file
|
| 201 |
+
provider: main
|
| 202 |
+
- path: ''
|
| 203 |
+
schema: structured
|
| 204 |
+
provider: schema
|
| 205 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-09/18-33-56
|
| 206 |
+
choices:
|
| 207 |
+
hydra/env: default
|
| 208 |
+
hydra/callbacks: null
|
| 209 |
+
hydra/job_logging: default
|
| 210 |
+
hydra/hydra_logging: default
|
| 211 |
+
hydra/hydra_help: default
|
| 212 |
+
hydra/help: default
|
| 213 |
+
hydra/sweeper: basic
|
| 214 |
+
hydra/launcher: basic
|
| 215 |
+
hydra/output: default
|
| 216 |
+
verbose: false
|
outputs/2026-06-15/11-17-58/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,210 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.mode=RUN
|
| 114 |
+
task:
|
| 115 |
+
- algorithm.adv_estimator=grpo
|
| 116 |
+
- data.train_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet
|
| 117 |
+
- data.val_files=/tmp/anomseer_eval_smoke.parquet
|
| 118 |
+
- data.train_batch_size=16
|
| 119 |
+
- data.max_prompt_length=1024
|
| 120 |
+
- data.max_response_length=1024
|
| 121 |
+
- data.filter_overlong_prompts=True
|
| 122 |
+
- data.truncation=error
|
| 123 |
+
- data.image_key=images
|
| 124 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 125 |
+
- actor_rollout_ref.model.lora_rank=0
|
| 126 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 127 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 128 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 129 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 130 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 131 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=16
|
| 132 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 133 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 134 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 135 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 136 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 137 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 138 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 139 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 140 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 141 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 142 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 143 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 144 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 145 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 146 |
+
- actor_rollout_ref.rollout.n=5
|
| 147 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 148 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 149 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 150 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 151 |
+
- trainer.critic_warmup=0
|
| 152 |
+
- trainer.logger=['console']
|
| 153 |
+
- trainer.project_name=anomseer
|
| 154 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 155 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 156 |
+
- trainer.n_gpus_per_node=2
|
| 157 |
+
- trainer.nnodes=1
|
| 158 |
+
- trainer.save_freq=500
|
| 159 |
+
- trainer.test_freq=10
|
| 160 |
+
- trainer.val_only=True
|
| 161 |
+
- trainer.val_before_train=True
|
| 162 |
+
- trainer.resume_mode=disable
|
| 163 |
+
- trainer.total_epochs=1
|
| 164 |
+
- ts.use_sem_orth=True
|
| 165 |
+
- ts.adv_mix=0.3
|
| 166 |
+
- ts.similarity_method=ot
|
| 167 |
+
- ts.ot_eps=0.08
|
| 168 |
+
- ts.ot_n_iter=50
|
| 169 |
+
- data.val_batch_size=1
|
| 170 |
+
- trainer.test_freq=-1
|
| 171 |
+
job:
|
| 172 |
+
name: main_ppo
|
| 173 |
+
chdir: null
|
| 174 |
+
override_dirname: actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.001,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-4,actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2,actor_rollout_ref.actor.ppo_mini_batch_size=16,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.lora_alpha=16,actor_rollout_ref.model.lora_dropout=0.0,actor_rollout_ref.model.lora_rank=0,actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj',actor_rollout_ref.model.path=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface,actor_rollout_ref.model.use_remove_padding=False,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.enable_chunked_prefill=False,actor_rollout_ref.rollout.enforce_eager=False,actor_rollout_ref.rollout.free_cache_engine=False,actor_rollout_ref.rollout.gpu_memory_utilization=0.4,actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8,actor_rollout_ref.rollout.n=5,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.tensor_model_parallel_size=2,actor_rollout_ref.rollout.val_kwargs.do_sample=False,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.001,data.filter_overlong_prompts=True,data.image_key=images,data.max_prompt_length=1024,data.max_response_length=1024,data.train_batch_size=16,data.train_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet,data.truncation=error,data.val_batch_size=1,data.val_files=/tmp/anomseer_eval_smoke.parquet,trainer.critic_warmup=0,trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu,trainer.experiment_name=anomseer_rats_uni_2gpu,trainer.logger=['console'],trainer.n_gpus_per_node=2,trainer.nnodes=1,trainer.project_name=anomseer,trainer.resume_mode=disable,trainer.save_freq=500,trainer.test_freq=-1,trainer.test_freq=10,trainer.total_epochs=1,trainer.val_before_train=True,trainer.val_only=True,ts.adv_mix=0.3,ts.ot_eps=0.08,ts.ot_n_iter=50,ts.similarity_method=ot,ts.use_sem_orth=True
|
| 175 |
+
id: ???
|
| 176 |
+
num: ???
|
| 177 |
+
config_name: ppo_trainer
|
| 178 |
+
env_set: {}
|
| 179 |
+
env_copy: []
|
| 180 |
+
config:
|
| 181 |
+
override_dirname:
|
| 182 |
+
kv_sep: '='
|
| 183 |
+
item_sep: ','
|
| 184 |
+
exclude_keys: []
|
| 185 |
+
runtime:
|
| 186 |
+
version: 1.3.0
|
| 187 |
+
version_base: '1.3'
|
| 188 |
+
cwd: /mnt/share01/sqk/AnomSeer
|
| 189 |
+
config_sources:
|
| 190 |
+
- path: hydra.conf
|
| 191 |
+
schema: pkg
|
| 192 |
+
provider: hydra
|
| 193 |
+
- path: /mnt/share01/sqk/AnomSeer/verl/trainer/config
|
| 194 |
+
schema: file
|
| 195 |
+
provider: main
|
| 196 |
+
- path: ''
|
| 197 |
+
schema: structured
|
| 198 |
+
provider: schema
|
| 199 |
+
output_dir: /mnt/share01/sqk/AnomSeer/outputs/2026-06-15/11-17-58
|
| 200 |
+
choices:
|
| 201 |
+
hydra/env: default
|
| 202 |
+
hydra/callbacks: null
|
| 203 |
+
hydra/job_logging: default
|
| 204 |
+
hydra/hydra_logging: default
|
| 205 |
+
hydra/hydra_help: default
|
| 206 |
+
hydra/help: default
|
| 207 |
+
hydra/sweeper: basic
|
| 208 |
+
hydra/launcher: basic
|
| 209 |
+
hydra/output: default
|
| 210 |
+
verbose: false
|
outputs/2026-06-15/11-17-58/main_ppo.log
ADDED
|
File without changes
|
outputs/2026-06-15/11-25-36/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
train_files: /mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet
|
| 4 |
+
val_files: /tmp/anomseer_eval_smoke.parquet
|
| 5 |
+
prompt_key: prompt
|
| 6 |
+
max_prompt_length: 1024
|
| 7 |
+
max_response_length: 1024
|
| 8 |
+
train_batch_size: 16
|
| 9 |
+
val_batch_size: 1
|
| 10 |
+
return_raw_input_ids: false
|
| 11 |
+
return_raw_chat: false
|
| 12 |
+
shuffle: true
|
| 13 |
+
filter_overlong_prompts: true
|
| 14 |
+
truncation: error
|
| 15 |
+
image_key: images
|
| 16 |
+
actor_rollout_ref:
|
| 17 |
+
hybrid_engine: true
|
| 18 |
+
model:
|
| 19 |
+
path: /mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 20 |
+
external_lib: null
|
| 21 |
+
override_config: {}
|
| 22 |
+
enable_gradient_checkpointing: true
|
| 23 |
+
use_remove_padding: false
|
| 24 |
+
lora_rank: 0
|
| 25 |
+
lora_alpha: 16
|
| 26 |
+
lora_dropout: 0.0
|
| 27 |
+
lora_target_modules: q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj
|
| 28 |
+
actor:
|
| 29 |
+
strategy: fsdp
|
| 30 |
+
ppo_mini_batch_size: 16
|
| 31 |
+
ppo_micro_batch_size: null
|
| 32 |
+
ppo_micro_batch_size_per_gpu: 2
|
| 33 |
+
use_dynamic_bsz: false
|
| 34 |
+
ppo_max_token_len_per_gpu: 16384
|
| 35 |
+
grad_clip: 1.0
|
| 36 |
+
clip_ratio: 0.2
|
| 37 |
+
entropy_coeff: 0.001
|
| 38 |
+
use_kl_loss: true
|
| 39 |
+
use_torch_compile: true
|
| 40 |
+
kl_loss_coef: 0.001
|
| 41 |
+
kl_loss_type: low_var_kl
|
| 42 |
+
ppo_epochs: 1
|
| 43 |
+
shuffle: false
|
| 44 |
+
ulysses_sequence_parallel_size: 1
|
| 45 |
+
optim:
|
| 46 |
+
lr: 0.0001
|
| 47 |
+
lr_warmup_steps: -1
|
| 48 |
+
lr_warmup_steps_ratio: 0.0
|
| 49 |
+
min_lr_ratio: null
|
| 50 |
+
warmup_style: constant
|
| 51 |
+
total_training_steps: -1
|
| 52 |
+
fsdp_config:
|
| 53 |
+
wrap_policy:
|
| 54 |
+
min_num_params: 0
|
| 55 |
+
param_offload: false
|
| 56 |
+
optimizer_offload: false
|
| 57 |
+
fsdp_size: -1
|
| 58 |
+
ref:
|
| 59 |
+
fsdp_config:
|
| 60 |
+
param_offload: true
|
| 61 |
+
wrap_policy:
|
| 62 |
+
min_num_params: 0
|
| 63 |
+
log_prob_micro_batch_size: null
|
| 64 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 65 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 66 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 67 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 68 |
+
rollout:
|
| 69 |
+
name: vllm
|
| 70 |
+
temperature: 1.0
|
| 71 |
+
top_k: -1
|
| 72 |
+
top_p: 1
|
| 73 |
+
use_fire_sampling: false
|
| 74 |
+
prompt_length: ${data.max_prompt_length}
|
| 75 |
+
response_length: ${data.max_response_length}
|
| 76 |
+
dtype: bfloat16
|
| 77 |
+
gpu_memory_utilization: 0.4
|
| 78 |
+
ignore_eos: false
|
| 79 |
+
enforce_eager: false
|
| 80 |
+
free_cache_engine: false
|
| 81 |
+
load_format: dummy_dtensor
|
| 82 |
+
tensor_model_parallel_size: 2
|
| 83 |
+
max_num_batched_tokens: 8192
|
| 84 |
+
max_model_len: null
|
| 85 |
+
max_num_seqs: 1024
|
| 86 |
+
log_prob_micro_batch_size: null
|
| 87 |
+
log_prob_micro_batch_size_per_gpu: 8
|
| 88 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 89 |
+
log_prob_max_token_len_per_gpu: ${actor_rollout_ref.actor.ppo_max_token_len_per_gpu}
|
| 90 |
+
disable_log_stats: true
|
| 91 |
+
enable_chunked_prefill: false
|
| 92 |
+
do_sample: true
|
| 93 |
+
'n': 5
|
| 94 |
+
val_kwargs:
|
| 95 |
+
top_k: -1
|
| 96 |
+
top_p: 1.0
|
| 97 |
+
temperature: 0.4
|
| 98 |
+
'n': 1
|
| 99 |
+
do_sample: false
|
| 100 |
+
critic:
|
| 101 |
+
strategy: fsdp
|
| 102 |
+
optim:
|
| 103 |
+
lr: 1.0e-05
|
| 104 |
+
lr_warmup_steps_ratio: 0.0
|
| 105 |
+
min_lr_ratio: null
|
| 106 |
+
warmup_style: constant
|
| 107 |
+
total_training_steps: -1
|
| 108 |
+
model:
|
| 109 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 110 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 111 |
+
override_config: {}
|
| 112 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 113 |
+
enable_gradient_checkpointing: true
|
| 114 |
+
use_remove_padding: false
|
| 115 |
+
fsdp_config:
|
| 116 |
+
param_offload: false
|
| 117 |
+
optimizer_offload: false
|
| 118 |
+
wrap_policy:
|
| 119 |
+
min_num_params: 0
|
| 120 |
+
fsdp_size: -1
|
| 121 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 122 |
+
ppo_micro_batch_size: null
|
| 123 |
+
ppo_micro_batch_size_per_gpu: null
|
| 124 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 125 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 126 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 127 |
+
ppo_max_token_len_per_gpu: 32768
|
| 128 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 129 |
+
ulysses_sequence_parallel_size: 1
|
| 130 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 131 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 132 |
+
grad_clip: 1.0
|
| 133 |
+
cliprange_value: 0.5
|
| 134 |
+
reward_model:
|
| 135 |
+
enable: false
|
| 136 |
+
strategy: fsdp
|
| 137 |
+
model:
|
| 138 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 139 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 140 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 141 |
+
use_remove_padding: false
|
| 142 |
+
fsdp_config:
|
| 143 |
+
wrap_policy:
|
| 144 |
+
min_num_params: 0
|
| 145 |
+
param_offload: false
|
| 146 |
+
fsdp_size: -1
|
| 147 |
+
micro_batch_size: null
|
| 148 |
+
micro_batch_size_per_gpu: null
|
| 149 |
+
max_length: null
|
| 150 |
+
ulysses_sequence_parallel_size: 1
|
| 151 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 152 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 153 |
+
reward_manager: naive
|
| 154 |
+
custom_reward_function:
|
| 155 |
+
path: null
|
| 156 |
+
name: compute_score
|
| 157 |
+
algorithm:
|
| 158 |
+
gamma: 1.0
|
| 159 |
+
lam: 1.0
|
| 160 |
+
adv_estimator: grpo
|
| 161 |
+
kl_penalty: kl
|
| 162 |
+
kl_ctrl:
|
| 163 |
+
type: fixed
|
| 164 |
+
kl_coef: 0.001
|
| 165 |
+
trainer:
|
| 166 |
+
balance_batch: true
|
| 167 |
+
total_epochs: 1
|
| 168 |
+
total_training_steps: null
|
| 169 |
+
project_name: anomseer
|
| 170 |
+
experiment_name: anomseer_rats_uni_2gpu
|
| 171 |
+
logger:
|
| 172 |
+
- console
|
| 173 |
+
val_generations_to_log_to_wandb: 0
|
| 174 |
+
nnodes: 1
|
| 175 |
+
n_gpus_per_node: 2
|
| 176 |
+
save_freq: 500
|
| 177 |
+
resume_mode: disable
|
| 178 |
+
resume_from_path: false
|
| 179 |
+
test_freq: -1
|
| 180 |
+
critic_warmup: 0
|
| 181 |
+
default_hdfs_dir: null
|
| 182 |
+
remove_previous_ckpt_in_save: false
|
| 183 |
+
del_local_ckpt_after_load: false
|
| 184 |
+
default_local_dir: checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 185 |
+
val_only: true
|
| 186 |
+
val_before_train: true
|
| 187 |
+
ts:
|
| 188 |
+
adv_mix: 0.3
|
| 189 |
+
use_sem_orth: true
|
| 190 |
+
similarity_method: ot
|
| 191 |
+
clip_temperature: 1.0
|
| 192 |
+
clip_pooling: mean
|
| 193 |
+
ot_eps: 0.08
|
| 194 |
+
ot_n_iter: 50
|
outputs/2026-06-15/11-25-36/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- data.train_files=/mnt/share01/sqk/AnomSeer/data/rats_uni_processed/train_quarter.parquet
|
| 3 |
+
- data.val_files=/tmp/anomseer_eval_smoke.parquet
|
| 4 |
+
- data.train_batch_size=16
|
| 5 |
+
- data.max_prompt_length=1024
|
| 6 |
+
- data.max_response_length=1024
|
| 7 |
+
- data.filter_overlong_prompts=True
|
| 8 |
+
- data.truncation=error
|
| 9 |
+
- data.image_key=images
|
| 10 |
+
- actor_rollout_ref.model.path=/mnt/share01/sqk/AnomSeer/checkpoints/anomseer/anomseer_rats_uni_full_20260612_184427_exp3/global_step_1891/actor/huggingface
|
| 11 |
+
- actor_rollout_ref.model.lora_rank=0
|
| 12 |
+
- actor_rollout_ref.model.lora_alpha=16
|
| 13 |
+
- actor_rollout_ref.model.lora_dropout=0.0
|
| 14 |
+
- actor_rollout_ref.model.lora_target_modules='q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj'
|
| 15 |
+
- actor_rollout_ref.actor.optim.lr=1e-4
|
| 16 |
+
- actor_rollout_ref.model.use_remove_padding=False
|
| 17 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=16
|
| 18 |
+
- actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=2
|
| 19 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 20 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.001
|
| 21 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 22 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 23 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 25 |
+
- actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=8
|
| 26 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=2
|
| 27 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 28 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.4
|
| 29 |
+
- actor_rollout_ref.rollout.enable_chunked_prefill=False
|
| 30 |
+
- actor_rollout_ref.rollout.enforce_eager=False
|
| 31 |
+
- actor_rollout_ref.rollout.free_cache_engine=False
|
| 32 |
+
- actor_rollout_ref.rollout.n=5
|
| 33 |
+
- actor_rollout_ref.rollout.val_kwargs.do_sample=False
|
| 34 |
+
- actor_rollout_ref.ref.log_prob_micro_batch_size_per_gpu=8
|
| 35 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 36 |
+
- algorithm.kl_ctrl.kl_coef=0.001
|
| 37 |
+
- trainer.critic_warmup=0
|
| 38 |
+
- trainer.logger=['console']
|
| 39 |
+
- trainer.project_name=anomseer
|
| 40 |
+
- trainer.experiment_name=anomseer_rats_uni_2gpu
|
| 41 |
+
- trainer.default_local_dir=checkpoints/anomseer/anomseer_rats_uni_2gpu
|
| 42 |
+
- trainer.n_gpus_per_node=2
|
| 43 |
+
- trainer.nnodes=1
|
| 44 |
+
- trainer.save_freq=500
|
| 45 |
+
- trainer.test_freq=10
|
| 46 |
+
- trainer.val_only=True
|
| 47 |
+
- trainer.val_before_train=True
|
| 48 |
+
- trainer.resume_mode=disable
|
| 49 |
+
- trainer.total_epochs=1
|
| 50 |
+
- ts.use_sem_orth=True
|
| 51 |
+
- ts.adv_mix=0.3
|
| 52 |
+
- ts.similarity_method=ot
|
| 53 |
+
- ts.ot_eps=0.08
|
| 54 |
+
- ts.ot_n_iter=50
|
| 55 |
+
- data.val_batch_size=1
|
| 56 |
+
- trainer.test_freq=-1
|
outputs/2026-06-15/11-25-36/main_ppo.log
ADDED
|
File without changes
|