pytorch
ghyz
learning to the deep
展开
-
Pytorch使用collate_fn拼接维度不同的数据LSTM
DataLoader有一个参数collate_fn,这个参数接收自定义`collate`函数,这个函数在数据加载(即通过Dataloader取一个batch数据)之前,定义对每个batch数据的处理方式。原创 2022-09-21 15:10:31 · 1061 阅读 · 1 评论 -
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
在使用pytorch训练完模型后,再进行test出现`RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR`错误。奇怪的是,之前测试的时候都能正常运行,偏偏现在不行,代码数据没有任何变化。原创 2022-06-17 19:17:45 · 6362 阅读 · 2 评论