场景:
使用多线程方式执行推理任务,调用的推理任务主函数作为多线程,推理任务的处理过程比较长,其中涉及到数据的前处理、推理、后处理等。
其中在推理计算精度时,需要将推理结果写入到一个文件中,问题来了,多个线程中如果同一个模型的话,会向同一个文件写数据,会出现数据冲突报错的问题。
参考了网上的建议,在出错那个方法中,尝试使用锁的方式,目前看不再报错了,
加入了这两行,其他不变
1、方法1 :后来有些问题
2、方法2:
验证仍然有些问题
3、方法3:
验证中
参考博文:
Python中的多线程编程,线程安全与锁(一) - 青山牧云人 - 博客园