LLaMA-Factory如何对Tokenization步骤提速

文章讨论了在处理大数据量时,Tokenization步骤耗时过长的问题。由于Tokenization不依赖GPU,作者提出了通过设置preprocessing_num_workers为多线程并行以及限制max_samples数量来提升CPU并发效率的方法。
Step3-VL-10B-Base阶跃星辰轻量级多模态基础模型

Step3-VL-10B-Base阶跃星辰轻量级多模态基础模型

Llama
文本生成
图像识别

视觉理解:图像识别、OCR、实体定位、计数、空间理解、GUI 交互 多模态推理:看图问答、图文理解、复杂逻辑推理(STEM、数学、代码)

问题

数据量稍微大一些,在运行这个步骤“Running tokenizer on dataset”就要等半小时以上,然后才知道后续是否会报错。Tokenization步骤看上去并没有使用到GPU,只是在CPU上运行的。
是否有什么方法对这个步骤进行加速呢,比如多CPU并发, 具体要如何做到Tokenization步骤提速呢?

方案

tokenization 无法用 GPU 运行。
多 CPU 并发通过设置 preprocessing_num_workers=16 来实现。
另外可以通过设置 max_samples 来限制 debug 过程中的样本数量。

如下这样设置就能提速

    --preprocessing_num_workers 16 \

您可能感兴趣的与本文相关的镜像

Step3-VL-10B-Base阶跃星辰轻量级多模态基础模型

Step3-VL-10B-Base阶跃星辰轻量级多模态基础模型

Llama
文本生成
图像识别

视觉理解:图像识别、OCR、实体定位、计数、空间理解、GUI 交互 多模态推理:看图问答、图文理解、复杂逻辑推理(STEM、数学、代码)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值