![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
. 小王同学.
这个作者很懒,什么都没留下…
展开
-
CUDA out of memory.最简单粗暴的解决办法
最近在跑一个神经网络时,出现:CUDA out of memory.的错误,可能是因为GPU内存不足的原因,最简单粗暴的方法就是将batch_size减小,但是会变慢。原创 2020-09-19 09:23:34 · 2749 阅读 · 2 评论 -
ONNX简介
简介Open Neural Network Exchange(ONNX,开放神经网络交换)格式,是一个用于表示深度学习模型的标准,可使模型在不同框架之间进行转移。ONNX是一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型。它使得不同的人工智能框架(如Pytorch, MXNet)可以采用相同格式存储模型数据并交互。 ONNX的规范及代码主要由微软,亚马逊 ,Facebook 和 IBM 等公司共同开发,以开放源代码的方式托管在Github上。目前官方支持加载ONNX模型并进行推理的深度学习转载 2020-09-13 09:30:21 · 1152 阅读 · 0 评论 -
Pycharm中对多行程序同时注释
最近在B站发现有的up主可以一下注释掉多行程序,感觉很实用,所以这里介绍一下:只需选中需要注释的行,并按“ctrl+/"键,即可对这些行进行注释,再次按"ctrl+/"即可取消注释。原创 2020-09-05 21:30:47 · 2723 阅读 · 3 评论 -
OpenCV中的resize函数与Pytorch中的transforms.Resize函数的区别
在OpenCV中,resize函数的语法格式为:resize(img1,img2,Size(width,height))注意:Size()里面是宽在前,而高在后!!!在PyTorch中,transforms.Resize()函数的语法格式为:transforms.Resize(height,width)注意Resize()里面是高在前,而宽在后!!!由于之前没有搞清楚这两个,导致在使用C++部署libtorch模型时,出现了预测严重失败的情况!...原创 2020-08-18 09:46:26 · 1870 阅读 · 0 评论