Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

量化训练qwen2-72b-instruct 的权重突入中断 #1151

Open
lxb0425 opened this issue Jun 17, 2024 · 1 comment
Open

量化训练qwen2-72b-instruct 的权重突入中断 #1151

lxb0425 opened this issue Jun 17, 2024 · 1 comment

Comments

@lxb0425
Copy link

lxb0425 commented Jun 17, 2024

Describe the bug
What the bug is, and how to reproduce, better with screenshots(描述bug以及复现过程,最好有截图)
使用2*A100对wen2-72b-instruct训练,完成后执行量化4位命令,一直无故中断,同步看显存还有很多
执行命令CUDA_VISIBLE_DEVICES=0,1 swift export --ckpt_dir '/workspace/sdata/qwen2-72b-instruct/v1-20240613-072850/checkpoint-132/' --merge_lora true --quant_bits 4 --load_dataset_config true --quant_method gptq
image

Your hardware and system info
Write your system info like CUDA version/system/GPU/torch version here(在这里给出硬件信息和系统信息,如CUDA版本,系统,GPU型号和torch版本等)

cuda=12.1
Additional context
Add any other context about the problem here(在这里补充其他信息)

@tastelikefeet
Copy link
Collaborator

增加OMP_NUM_THREADS=14环境变量试一下

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants