- 博客(13)
- 收藏
- 关注
原创 TypeError pybind11::init(): factory function returned nullptr
最后解决方法是重新安装 `nvidiia-tensorrt==8.4.0.6` 、如果不指定版本的话,默认是99.0.0.在使用yolov5转engine的时候报错。
2023-03-27 14:09:32
4232
2
原创 pytorch实现卷积层和BN层融合
批归一化数据的规范化也即(x−mean(x)/var(x))(x-mean(x)/var(x))(x−mean(x)/var(x)) 可以将数据的不同特征规范到均值为0,方差相同的标准正态分布,这就可以避免大数造成得数据溢出。可以使网络以更高得学习率更快得收敛。但是随着网络深度得增加,数据得分布会逐渐偏离标准正态分布,由于梯度得反向传播是不同层得梯度不断相乘得结果,数据过小或过大容易造成梯度消失或梯度爆炸。为了解决上述问题,提出了对网络层得中间输出进行规范化得方法,也就是批规范化(Batch Nor
2022-05-20 11:39:45
2745
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人