ymcui/Chinese-LLaMA-Alpaca-2

多卡训练卡在加载模型

ymourenya opened this issue · 7 comments

提交前必须检查以下项目

  • 请确保使用的是仓库最新代码(git pull),一些问题已被解决和修复。
  • 我已阅读项目文档FAQ章节并且已在Issue中对问题进行了搜索,没有找到相似问题和解决方案。
  • 第三方插件问题:例如llama.cppLangChaintext-generation-webui等,同时建议到对应的项目中查找解决方案。

问题类型

其他问题

基础模型

Chinese-LLaMA-2 (7B/13B)

操作系统

None

详细描述问题

# 请在此处粘贴运行代码(请粘贴在本代码块里)

依赖情况(代码类问题务必提供)

# 请在此处粘贴依赖情况(请粘贴在本代码块里)

运行日志或截图

torchrun --nnodes 1 --nproc_per_node 4 run_clm_pt_with_peft.py
你好,我把gpu的数量设置为4,采用单机多卡进行训练,为什么会卡在加载模型,我单卡进行训练没有这个问题
图片

注释掉L549试试

你好,还是不行

注释掉L549试试

你好,我两张gpu也能训练,但是再往上就不行了,我用的A40

注释掉L549试试

还有就是能问一下你们是用的什么GPU进行预训练的吗

我们也是A40。L244改为False呢?双卡既然能跑,分布式代码应该是没有问题的。

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your consideration.

Closing the issue, since no updates observed. Feel free to re-open if you need any further assistance.