PyTorch
文章平均质量分 70
牧羊女说
有马甲线的女程序猿、攻城狮、加班狗~
展开
-
LSTM学习笔记
LSTM模块的Pytorch实现原创 2024-01-20 18:35:40 · 1080 阅读 · 0 评论 -
OCR识别网络CRNN理解与Pytorch实现
CRNN论文理解及代码实现原创 2024-01-19 18:28:44 · 1900 阅读 · 0 评论 -
PyTorch随机数生成:torch.rand,torch.randn,torch.randind,torch.rand_like
pytorch中几个随机函数的定义和用法整理。原创 2023-12-22 16:40:25 · 4621 阅读 · 0 评论 -
Pytorch图像模型转ONNX后出现色偏问题
本篇记录一次从Pytorch图像处理模型转换成ONNX模型之后,在推理过程中出现了明显色偏问题的解决过程。原创 2023-11-03 18:43:07 · 699 阅读 · 0 评论 -
视频去噪网络BSVD的实现
前些天写了视频去噪网络BSVD论文的理解,详情请点击,这两个星期动手实践了一下,本篇就来记录一下这个模型的实现。这个网络的独特之处在于,它的训练和推理在实现上有所差别。在训练阶段,其使用了TSM(Time Shift Module)结构,而在推理时则使用了BBB(Bidirectional Buffer Block)结构。训练时,网络是一个MIMO(多输入多输出)形式,而在推理时,则将其设计成了单输入、单输出的流式形式。原创 2023-10-23 21:05:07 · 783 阅读 · 1 评论 -
Pytorch QAT for UNet
模型的QAT过程原创 2023-06-20 14:53:24 · 1710 阅读 · 15 评论 -
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
发现打印出来是CPU,说明模型还在CPU上,没有load到cuda。看意思应该是有些tensor在cuda上,有些tensor在cpu上,需要将两者统一起来。原创 2023-06-16 18:37:51 · 955 阅读 · 0 评论 -
“Torch not compiled with CUDA enabled“
解决"Torch not compiled with CUDA enabled"错误的过程。原创 2023-01-19 20:19:16 · 1025 阅读 · 0 评论 -
AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘
Jetson平台上安装PyTorch的问题定位过程。原创 2022-07-05 16:29:47 · 4775 阅读 · 0 评论 -
RuntimeError: no valid convolution algorithms available in CuDNN
使用PyTorch遇到的问题及解决方法。原创 2022-07-01 19:35:53 · 5690 阅读 · 5 评论 -
YOLOX加强特征提取网络Panet分析
YOLOX加强特征提取网络PANET的结构分析和代码实现。原创 2022-06-28 17:03:59 · 1561 阅读 · 0 评论 -
YOLOX backbone——CSPDarknet的实现
YOLOX的主干网络分析原创 2022-06-22 21:27:48 · 8961 阅读 · 1 评论 -
PyTorch学习笔记:使用不同设备(CPU/GPU)对模型进行保存和加载
本篇其实与PyTorch学习笔记:使用state_dict来保存和加载模型是高度关联的,之所以单独拎出来写,主要是想突出它的重要性。首先来描述一个本人实际遇到的问题:首先在GPU服务器上训练了一个ResNet34的模型,然后将该模型在本人PC机(没有GPU)上进行推理,模型加载代码如下:# load model weightsweights_path = "./resNet34.pth"assert os.path.exists(weights_path), "file: '{}' dos原创 2022-05-09 17:49:59 · 2830 阅读 · 9 评论 -
PyTorch学习笔记:使用state_dict来保存和加载模型
1. state_dict简介state_dict是Python的字典对象,可用于保存模型参数、超参数以及优化器(torch.optim)的状态信息。需要注意的是,只有具有可学习参数的层(如卷积层、线性层等)才有state_dict。下面就拿官方教程中的一个小示例来说明state_dict的使用:import torchimport torch.nn as nnimport torch.optim as optim# 定义模型class TheModelClass(nn.Module原创 2022-05-09 16:47:12 · 16174 阅读 · 2 评论 -
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memor
在Docker容器中利用Pytorch训练模型的时候,出现了题目中的错误:ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm).ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm).原创 2022-05-07 16:41:45 · 9536 阅读 · 8 评论