pytorch
野鹤无粮
Timing...
展开
-
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)【pytorch err】
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling 'cublasCreate(handle)问题描述使用 pytorch 跑程序 环境没变,换了代码就报了这个错误 RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling 'cublasCreate(handle),尝试使用CPU运行,发现是可以跑通的,没有提示什么问题,但是换成GPU跑就不行了原创 2022-05-20 16:31:38 · 2709 阅读 · 2 评论 -
pytorch 实现多维FFT
pytorch 实现多维FFT, 即fftn文章目录前言前言根据官方文档, pytorch 中已经有多维fft的实现,不过要先 import torch.fft代码示例1import torch.fft as fft# 实现一维fftfft.fft(input, n=None, dim=-1, norm=None)# 实现n维fftfft.fftn(input, s=None, dim=None, norm=None) input: (张量类型)输入张量n: (int 类型翻译 2021-10-21 21:07:23 · 2749 阅读 · 0 评论 -
pytorch 模型容器 - 模块化构建深度学习网络
pytorch 模型容器 - 模块化构建深度学习网络目录pytorch 模型容器 - 模块化构建深度学习网络pytorch 模型容器总结一、nn.Sequential二、nn.ModuleList三、nn.ModuleDict后记pytorch 模型容器总结pytorch 提供的模型容器包括:nn.Sequential:按顺序包含多个网络层nn.ModuleList:类似列表(list)的形式包含多个网络层nn.ModuleDict:类似字典(dict)的形式包含多个网络层一、nn.Se原创 2021-09-21 18:16:14 · 446 阅读 · 0 评论 -
小错误记录:‘Loss‘ object has no attribute ‘_forward_pre_hooks
记录一个小错误:使用pytorch 自定义loss时出现了这样的错误:torch.nn.modules.module.ModuleAttributeError: ‘Loss’ object has no attribute ‘_forward_pre_hooks’目录出现原因总结出现原因写 super 继承时 ,忘记加self# 原代码 未加self class LossSSIM(nn.Module): def __init__(self, k1=0.01, k2=0.03, L=原创 2021-09-15 21:32:35 · 1784 阅读 · 1 评论