会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Kurie
鹤鸣于九皋,声闻于野。
博客园
首页
新随笔
联系
订阅
管理
2023年12月3日
同一个train.py,没改任何代码,一会能跑,一会不能跑
摘要: 问题描述 跑一个模型,加载预训练权重,只进行推理,没有修改任何代码,刚开始直接运行可以得到推理结果,然后再跑,程序又卡住了;然后我使用调试,一会能跑,一会跑到第五六个batch时,又卡住了。 解决方法 修改dataloader的num_workers。我程序原来workers设为的8,后面改为4就可
阅读全文
posted @ 2023-12-03 14:18 Kurie
阅读(33)
评论(0)
推荐(0)
编辑
公告