![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 71
gangeqian2
欢迎来到我的博客,共同学习
展开
-
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memor
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm). 可以看到/dev/shm只有64M, 但是data_loader设置的num_works较多,且其通过共享内存协作,导致内存不足。 解决方法: (1) num_workers=0 (注意,设置为1不起作用) (2)docker 容易共享内存加大: 重启后:...原创 2022-06-17 14:25:39 · 1009 阅读 · 0 评论 -
onnx 模型转换与 onnxruntime 和caffe2 推理速度比较
onnx 模型转换与 onnxruntime 和caffe2 推理速度比较背景1、模型转换2、PC端运行onnx模型 背景 pytorch 模型通常包括网络结构.py文件和模型参数文件.pth,通常需要转换为onnx格式再向其他终端或移动端部署(onnx模型一般用于中间部署阶段,相当于翻译的作用)。 移动端通常使用ncnn、mnn、Caffe2进行inference,实现实时计算。 本文展示pytorch转onnx的模型在PC端 以CPU方式运行。 1、模型转换 1.1. pytorch 模型转onnx原创 2021-01-26 18:04:38 · 1861 阅读 · 1 评论