![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
粼粼淇
淇水汤汤,波光粼粼。岸芷汀兰,郁郁青青。
从事计算机视觉,机器学习等方面算法的工作,爱好数学、算法、编程、中国古典文学、哲学,喜欢旅游,爱思考、爱推理;追求卓越、追求极致。
展开
-
libtorch中cat函数的使用
在使用libtorch中经常用到vector和cat使用的情况,在此写了几个例子;需要注意的是,cat拼接tensor时必须时连续的tensor。原创 2022-12-10 20:55:44 · 505 阅读 · 0 评论 -
pytorch官方DDP示例中疑惑记录
官网上有关DDP的示例比较简短,说实话不太实用,但即使这样,第一次看这个还是有好些疑惑,在此把示例中,我当时的不解,经过查找和实验,做些注解.DDP讲解较好的参考文章:1 https://zhuanlan.zhihu.com/p/1784027982 https://zhuanlan.zhihu.com/p/76638962...原创 2022-06-30 21:52:45 · 669 阅读 · 0 评论 -
加载yolov5 模型到cuda0和加载resnet50模型到cuda1出错处理
先说下我工程:我需要使用yolov5的模型给出一张图中的检测结果,然后把这些检测结果再次送到resnet50的分类器中进行分类,刚开始我把两个模型都加载到了cuda0,一切完美运行.为了加速,我把yolov5的模型加载到了cuda0上(说明下我有4块gpu),resnet50的分类器模型加载到了cuda1上.想法是美好的,但执行上述流程时出现了意想不到的错误:RuntimeError: CUDA error: invalid device ordinalCUDA kernel errors mig原创 2022-01-06 17:50:40 · 1629 阅读 · 0 评论 -
pytorch分布式训练示例小计
示例地址:https://pytorch.org/tutorials/intermediate/ddp_tutorial.html完整代码如下:import osimport sysimport tempfileimport torchimport torch.distributed as distimport torch.nn as nnimport torch.optim as optimimport torch.multiprocessing as mpfrom torch原创 2021-11-16 18:06:01 · 1397 阅读 · 0 评论 -
np.meshgrid和torch.meshgrid的区别
两者的区别通过一个测试显而易见:a=[1,2,3]b=[4,5]gridx,gridy=np.meshgrid(a,b)print('gridx=\n',gridx)print('gridy=\n',gridy)x=torch.tensor(a)y=torch.tensor(b)gridx, gridy =torch.meshgrid(x,y)print('gridx=\n', gridx)print('gridy=\n', gridy)输出结果:gridx=[[1 .原创 2021-09-24 11:47:17 · 927 阅读 · 0 评论