modelscope / modelscope/ms-swift

CHORD算法在Qwen2.5-VL-32B不work

Open
#6,358 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
15.7k
Forks
1.7k
Avg merge
1d 16h
Merged PRs (30d)
136

Description

grpo与cold start数据格式完全相同,json文件包含images、problem、answer、solution四个字段。cold start中带有think过程,grpo只有标签。但log中感觉sft过程没有学习,配置文件信息如下:

export LOG_LEVEL=DEBUG
MAX_PIXELS=200704
WANDB_API_KEY=97c188294999d80264588d15d62791cf5af3da6a
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
NPROC_PER_NODE=8
swift rlhf
--rlhf_type grpo
--model /group/40106/model_output/Qwen_96w_full/checkpoint-227652
--model_type qwen2_5_vl
--dataset /group/40106/data/antifraud_lvlm/rl_data/data/train_with_solution.json
--val_dataset /group/40106/data/antifraud_lvlm/rl_data/data/val_with_solution.json
--load_from_cache_file True
--torch_dtype bfloat16
--beta 0.0
--lora_rank 64
--max_pixels 200704
--generation_batch_size 32
--num_train_epochs 1
--max_length 4096
--per_device_train_batch_size 2
--per_device_eval_batch_size 1
--gradient_accumulation_steps 2
--chord_sft_per_device_train_batch_size 4
--chord_sft_dataset /group/40106/data/antifraud_lvlm/cold_start/data/cold_start_chord_with_solution.json
--chord_enable_phi_function True
--chord_mu_warmup_steps 0
--chord_mu_decay_steps 1000
--chord_mu_peak 0.95
--chord_mu_valley 0.05
--num_generations 8
--train_type lora
--freeze_vit true
--attn_impl flash_attention_2
--external_plugins examples/train/grpo/plugin/plugin.py
--reward_funcs antifraud_reward
--use_vllm true
--vllm_mode colocate
--vllm_gpu_memory_utilization 0.5
--vllm_max_model_len 8192
--vllm_max_lora_rank 64
--vllm_tensor_parallel_size 4
--max_completion_length 1024
--overlong_filter true
--offload_optimizer true
--offload_model true
--sleep_level 1
--save_steps 1000
--eval_steps 1000
--learning_rate 1e-6
--save_total_limit 2
--logging_steps 1
--warmup_ratio 0.05
--dataloader_num_workers 4
--deepspeed zero3
--move_model_batches 20
--use_liger_kernel false
--output_dir /group/40106/model_output_rl/qwen25_chord
--log_completions true

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reproducing the swift rlhf command with the provided Qwen2.5-VL configuration and inspect the training logs around the CHORD SFT stage. Compare the train_with_solution.json and cold_start_chord_with_solution.json inputs and review examples/train/grpo/plugin/plugin.py; done means identifying why the SFT component is not learning and documenting a verified fix.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.