摘要:
参考 pytorch中的forward()的使用与解释 - 云+社区 - 腾讯云 前言 最近在使用pytorch的时候,模型训练时,不需要使用forward,只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数 即: forward 的使用 class Module(nn.Mod 阅读全文
摘要:
import torch pthfile = r'E:/pycharm_code/unet-pytorch-main/unet-pytorch-main/logs/Epoch1-Total_Loss0.4523-Val_Loss0.0966.pth' net = torch.load(pthfile 阅读全文
摘要:
tensorRT踩坑日常之engine推理 再进行tensorRT进行推理之前,需要将训练好的模型转onnx再进行序列化生成engine,然后反序列化context推理 此文章是进行序列化生成engine和推理的,不知道如何生成engine和onnx的小伙伴可以参考另一篇博客 https://blo 阅读全文
摘要:
报错: pycuda._driver.LogicError: explicit_context_dependent failed: invalid device context - no currently active context? TensorRT的调试报错整理: 原因:pycuda.dri 阅读全文
摘要:
python中@是什么意思 通常我们所见的@符号是用于邮箱中,而在python中@符号也有着重要的作用:一个是表示修饰符,另一个则是表示矩阵乘法(不常用)。表示修饰符时,可以在模块或者类的定义层内对函数进行修饰。虽然表示矩阵乘法不常用,但是也是很好的一个方法。值得注意的是@符号的这两种含义,都是在p 阅读全文
摘要:
介绍 .wts是一种可解释性文本 对于一个例子 lenet5.wts 来说,这里我们展示他的一部分。 10 conv1.weight 150 be40ee1b bd20bab8 bdc4bc53 ....... conv1.bias 6 bd327058 ....... conv2.weight 2 阅读全文
摘要:
若使用tensorrt加载wts格式,需将模型训练的pt、pth、ckpt等格式权重转换为wts。 我只记录此代码,供读者使用。 ? def checkpint2wts(model, wts_file): ''' model:模型,需要权重 wts_file:保存wts权重路径,如result.wt 阅读全文
摘要:
yum 安装g++时,用yum list的程序名不是 g++,而是 gcc-c++,否则搜不到。 先list: #yum list gcc-c++ Loaded plugins: fastestmirror Loading mirror speeds from cached hostfile * b 阅读全文
摘要:
GitHub项目的右上角都会有三个按钮,分别是Watch、Star和Fork,Watch、Star和Fork是什么意思?码笔记来说说Watch、Star和Fork的意思以及如何使用: Watch、Star和Fork 每个GitHub项目的右上角都会有Watch、Star和Fork三个按钮,如下图所示 阅读全文
摘要:
你的数据处理影响整个训练速度,如果加上英伟达 DALI 库,处理速度比原生 PyTorch 也能快上四倍。 选自towardsdatascience,作者:Pieterluitjens,机器之心编译,参与:一鸣、嘉明、思。 深度学习的加速上,除了对训练和推理过程的加速之外,还有对数据加载和预处理阶段 阅读全文