草庐IT

Yolov5_DeepSort_Pytorch

全部标签

Pytorch固定随机种子&&复现模型

官网 Reproducibility—PyTorch1.11.0documentation在神经网络中,参数默认是进行随机初始化的。不同的初始化参数往往会导致模型的训练结果会存在一定的差异。当得到比较好的结果时我们通常希望这个结果是可以复现的,就需要保证每一次初始化的参数都不变,这就引入了随机种子。在PyTorch中,通过设置全局随机数种子可以实现这个目的。本文总结了PyTorch中固定随机种子的方法。一 训练过程的不确定性在训练过程中,若相同的数据数据集,相同的训练集、测试集划分方式,相同的权重初始化,但是每次训练结果不同,可能有以下几个原因:PyTorch、Python、Numpy中的随机

【pytorch】torch.cdist使用说明

使用说明torch.cdist的使用介绍如官网所示,它是批量计算两个向量集合的距离。其中,x1和x2是输入的两个向量集合。p默认为2,为欧几里德距离。它的功能上等同于scipy.spatial.distance.cdist(input,’minkowski’,p=p)如果x1的shape是[B,P,M],x2的shape是[B,R,M],则cdist的结果shape是[B,P,R]进一步的解释x1一般是输入矢量,而x2一般是码本。x2中所有的元素分别与x1中的每一个元素求欧几里德距离(当p默认为2时)如下面示例importtorchx1=torch.FloatTensor([0.1,0.2,0

【pytorch】torch.cdist使用说明

使用说明torch.cdist的使用介绍如官网所示,它是批量计算两个向量集合的距离。其中,x1和x2是输入的两个向量集合。p默认为2,为欧几里德距离。它的功能上等同于scipy.spatial.distance.cdist(input,’minkowski’,p=p)如果x1的shape是[B,P,M],x2的shape是[B,R,M],则cdist的结果shape是[B,P,R]进一步的解释x1一般是输入矢量,而x2一般是码本。x2中所有的元素分别与x1中的每一个元素求欧几里德距离(当p默认为2时)如下面示例importtorchx1=torch.FloatTensor([0.1,0.2,0

PyTorch model 返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()

PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()文章目录PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()前言一、model.modules(),model.children(),model.parameters()1.1model.modules()1.2model.named_modules()1.3model.children()

PyTorch model 返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()

PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()文章目录PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()前言一、model.modules(),model.children(),model.parameters()1.1model.modules()1.2model.named_modules()1.3model.children()

Docker配置深度学习pytorch gpu环境

一、docker的概念原理参考资料:https://zhuanlan.zhihu.com/p/81693661          https://www.zhihu.com/question/506867139          https://blog.csdn.net/weixin_44751294/article/details/123191429    看过上面两个链接,就对docker是什么作用,怎么用有个大概的了解。简单摘要一些对我来说比较重要的点如下:(1)docker可以让环境配置变得简单,在多人同时使用一台服务器或者时不时换电脑换机器所有的包时都很有用(2)docker可以看

Docker配置深度学习pytorch gpu环境

一、docker的概念原理参考资料:https://zhuanlan.zhihu.com/p/81693661          https://www.zhihu.com/question/506867139          https://blog.csdn.net/weixin_44751294/article/details/123191429    看过上面两个链接,就对docker是什么作用,怎么用有个大概的了解。简单摘要一些对我来说比较重要的点如下:(1)docker可以让环境配置变得简单,在多人同时使用一台服务器或者时不时换电脑换机器所有的包时都很有用(2)docker可以看

pytorch :OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading 【已解决】

OSError:[WinError1455]页面文件太小,无法完成操作。Errorloading"D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\torch\lib\cudnn_cnn_infer64_8.dll"oroneofitsdependencies.win11系统,运行出现上述错误 解决方法 1.降低Batch_Size大小:过大的Batch_Size(人工智能术语:指一次迭代过程送入的数据集样本量)显然需要更多的内存支持,出现“页面文件太小”,可尝试降低Batch_Size大小。2.调整虚拟内存可用硬盘空间大小:虚

pytorch :OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading 【已解决】

OSError:[WinError1455]页面文件太小,无法完成操作。Errorloading"D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\torch\lib\cudnn_cnn_infer64_8.dll"oroneofitsdependencies.win11系统,运行出现上述错误 解决方法 1.降低Batch_Size大小:过大的Batch_Size(人工智能术语:指一次迭代过程送入的数据集样本量)显然需要更多的内存支持,出现“页面文件太小”,可尝试降低Batch_Size大小。2.调整虚拟内存可用硬盘空间大小:虚

yolov7模型部署——环境搭建(python 导出onnx模型,c# OnnxRunTime-GPU版调用) 踩坑记录

一,导出onnx模块1.1运行python.\export.py导出onnx模块    yolov7提供了简单的点对点预测,不需要再重新写NMS,非常方便,于是当然采用--end2end方法啦,命令如下图: 运行时发现没装onnx,那就安装好了1.2安装onnxcondainstall-cconda-forgeonnx但是!又出现了这个问题原因是__int__()需要2到7个参数,但给了8个参数。查看代码export.py159行,确实是8个参数:修改为义下代码:  问题解决。2,.NETC#Microsoft.ML.OnnxRuntime.GPU版调用onnx模型2.1安装onnxrunti