摘要: 最近再用LSTM训练模型,有时候epoch进行几次后总会出现RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,心态都快崩了。记录一下自己踩过的坑吧,上网查了一下这个报错,大家说法不一,有人说是因为pytorch和cuda版本不对应,得重 阅读全文
posted @ 2021-01-09 21:49 正在学习的小陈 阅读(25292) 评论(1) 推荐(0) 编辑
摘要: import matplotlib.pyplot as pltimport numpy as npimport torchfrom torch import nnfrom torch.autograd import Variablefrom visdom import Visdomimport os 阅读全文
posted @ 2021-01-02 22:31 正在学习的小陈 阅读(297) 评论(0) 推荐(0) 编辑
摘要: 此案例参考python数据科学手册,属于分类的一种,觉得挺有意思的。上代码 from sklearn.datasets import load_digitsimport seaborn as sns#导入数据digits=load_digits()X=digits.datay=digits.targ 阅读全文
posted @ 2020-12-28 22:34 正在学习的小陈 阅读(439) 评论(0) 推荐(0) 编辑
摘要: 研究方向是数据驱动的产品剩余寿命预测,看了不少论文,搞了几个月也没做出成果。打算从最简单的线性回归拟合开始。话不多说,直接开始。 1.数据 用的西交轴承库数据35Hz12kN工况的bearing1-1,网上能找到数据库。采样频率是25.6KHZ,一分钟内只采集1.28秒也就是一分钟内采集32768个 阅读全文
posted @ 2020-12-24 20:02 正在学习的小陈 阅读(1806) 评论(1) 推荐(0) 编辑