01 2021 档案

摘要:最近再用LSTM训练模型,有时候epoch进行几次后总会出现RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,心态都快崩了。记录一下自己踩过的坑吧,上网查了一下这个报错,大家说法不一,有人说是因为pytorch和cuda版本不对应,得重 阅读全文
posted @ 2021-01-09 21:49 正在学习的小陈 阅读(26026) 评论(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 正在学习的小陈 阅读(321) 评论(0) 推荐(0)