- 博客(12)
- 收藏
- 关注
原创 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
2712
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
2787
原创 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
450
原创 小错误记录:‘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
1810
1
原创 用python 读取matlab 中的结构体 / 从numpy.void中提取数据
从numpy.void中提取数据目录从numpy.void中提取数据前言一、示例二、代码汇总前言用python通过scipy 库读取mat文件,如果读取的是结构体,那么就会得到numpy.void 类型的数据。一、示例要读取如图所示的mat文件中的数据用scipy.sio.loadmat读取后得到的是:def load_params(path,key_name): data = sio.loadmat(path) data = data[key_name] re
2021-08-22 21:20:55
4176
6
原创 用python 实现2D FFT
用python 实现2D FFT目录用python 实现2D FFT前言代码结果总结前言用python实现2D FFT,并绘制结果代码直接上代码import numpy as npfrom scipy.fftpack import fftshift,fftnimport matplotlib.pyplot as pltfrom matplotlib.pyplot import MultipleLocatorN = 128 # 采
2021-08-17 20:09:58
3309
原创 用 python 实现FFT,绘制频谱图
用 python 实现FFT关键词 :fft , scipy 库, fftshift ,单边谱,双边谱,频谱泄露目录用 python 实现FFT前言代码结果总结和讨论前言之前都是在matlab上实现FFT,现在因为需要,在python上进行实现,在此做一个记录。代码直接上代码import numpy as npfrom scipy.fftpack import fft, ifft,fftshiftimport matplotlib.pyplot as pltN = 1024
2021-08-12 17:05:16
32581
17
原创 pycharm 右键 deployment 灰色 无法点
pycharm 右键 deployment 灰色 无法点pycharm 正常配置可以使用后,修改了使用的服务器环境。然后就出现了deployment 显示灰色,不能正常同步的问题。解决办法:查看同步的路径是否一致菜单栏 Tools > deployment > configure删掉了一个...
2021-07-11 16:13:38
5654
2
原创 Matlab 程序(二)
Matlab 程序(二)使用数值积分方法编写一个函数计算椭球的体积和表面积,输入参数为椭球的三个轴。MATLAB 代码function [S,V]=A_9(a,b,c)fun1=@(A,B) a*b.*cos(A).*sqrt(sin(A).^2+(c/a.*cos(A).*cos(B)).^2+(c/b.*cos(A).*sin(B)).^2);S=8*quad2d(fun1,0,pi/2,0,pi/2); % 0到pi/2进行积分,用对称性直接乘8fun2=@(x,y,z)x-x+y-y+z
2021-04-04 16:32:09
1112
原创 Matlab程序 (一)
matlab程序(一)求取多峰函数在x^2 +(y/3)^2=1所描绘的平面曲线上的积分思路将曲线积分转化成一元积分代码如下:function=@(t) (3*(1-cos(t)).^2.*exp(-(cos(t).^2) - (3*sin(t)+1).^2) ... - 10*(cos(t)/5 - cos(t).^3 - 3*sin(t).^5).*exp(-cos(t).^2-3*sin(t).^2) ... - 1/3*exp(-(cos(t)+1).^2 - 3*sin(t
2020-12-17 21:28:40
278
原创 tensorflow2 class多输入的操作
tensorflow2 class多输入的操作:有时模型训练需要导入多路输入数据,在原先模型的基础上,可以进行如下改动:(以2通道输入数据和1通道输出结果为例) def call(self,inputs): x0,x1=inputs # 用x0 x1接收inputs 数据,即x0=x0_train,x1=x1_train model.fit([x0_train, x1_train], y_train,batch_size=128, epochs=5, validati
2020-11-27 16:58:31
512
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人