![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
神经网络加速
Ai_Taoism
这个作者很懒,什么都没留下…
展开
-
pytorch的Backward过程用时太长?
pytorch的Backward过程用时太长?问题描述解决方案问题描述使用pytorch对网络进行训练的时候遇到一个问题,forward阶段很快(只需要几毫秒),backward阶段却用时很长(需要十多秒)。导致这个问题的原因很容易被大家忽视,而且网上基本上没有直接的解决方案,经过一天的折腾,总算把导致这个问题的原因搞清楚了。解决方案导致这个问题的原因在于训练数据的浅拷贝,由于backward过程中的梯度是和模型推理过程中的张量相关的,如果这些张量在被模型使用之前没有被深拷贝,意味着backward原创 2022-05-19 15:59:59 · 3147 阅读 · 7 评论 -
用pytorch计算神经网络模型需要占用的显存
用pytorch计算神经网络模型需要占用的显存#--------------------------------------------------------------------##作用:计算模型需要占用的显存,方便知道显卡够不够用#使用方法:将模型初始化之后,传入Calculate_gpu_memory()即可#--------------------------------------------------------------------#import torchimport原创 2021-09-28 19:46:23 · 2219 阅读 · 0 评论 -
用deepstream-test3跑yolov5,并对小汽车(car)进行流量统计
用deepstream-test3跑yolov5,并对小汽车car进行流量统计前言项目效果项目地址前言本文的项目是在Nvidia jetson上部署yolov5算法,并实现自定义功能(如对车辆进行流量统计)。在这里我是在deepstream-test3源码的基础上,对gstreamer进行更改,加入自定义功能,最后编译实现。项目效果见:https://www.bilibili.com/video/BV1Xh41187tQ/项目地址见:https://github.com/dongdongdong原创 2021-07-01 16:38:00 · 888 阅读 · 5 评论 -
一种可以加速深度学习推理的目标跟踪算法
自己发明的超实用目标检测算法前言算法原理算法实现前言所谓目标检测,就是在视频流的第一帧里面找到目标,标记目标,然后在后面的每一帧都准确找到目标的位置。算法原理算法实现原创 2020-12-30 23:26:39 · 430 阅读 · 0 评论