摘要: 问题描述 跑一个模型,加载预训练权重,只进行推理,没有修改任何代码,刚开始直接运行可以得到推理结果,然后再跑,程序又卡住了;然后我使用调试,一会能跑,一会跑到第五六个batch时,又卡住了。 解决方法 修改dataloader的num_workers。我程序原来workers设为的8,后面改为4就可 阅读全文
posted @ 2023-12-03 14:18 Kurie 阅读(33) 评论(0) 推荐(0) 编辑