python及pytorch异常错误处理
关注
分享
扫一扫
记录编程过程中遇到的python及pytorch异常错误以及处理方式
蓝海渔夫
主要方向:图像分类，图像分割；兴趣:生成式AI；下半年目标两千粉
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
fc层报错RuntimeError: mat1 and mat2 shapes cannot be multiplied (65536x1 and 512x1000)
我这儿报错的原因是将第一层全连接层替换成了全局池化，但是在第二层忘记修改增nn.Flatten()。导致全局池化出来的结果依然是二维的。提供一种在fc层报错的情况。加上图中选中的那行就行了。原创 2024-02-19 15:22:17 · 810 阅读 · 1 评论 -
pytorch报错：RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)
错误信息中的“Assertion `t >= 0 && t < n_classes` failed.”这句提示显示错误可能和number of classes有关。我在跑cifar10类别时，创建网络部分把num_classes设置成了6。这个报错看似是CUDA的问题，其实并不是。检查代码发现，我的代码设置的分类类别错了。原创 2024-03-07 21:47:02 · 1425 阅读 · 0 评论 -
复现r2unet或rnn过程中报错RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
今天尝试用pytorch复现了r2unet分割模型，这个模型在UNet中加入了recurrent block，也就是循环网络模块。看上去像是cudnn版本问题，但是其实是循环网络模块需要的显存太大了。我把batch size从20调到10之后，再次报错。这次的报错就很明显了，说明了显存占用问题。我再次将batch size降到5.原创 2024-02-28 14:31:09 · 444 阅读 · 1 评论 -
pytorch运行时tensorboard报错TypeError: Descriptors cannot be created directly.
这个错误主要是tensorboard版本的问题。解决办法是卸载tensorboard后重新安装。原创 2024-03-07 21:40:26 · 1002 阅读 · 0 评论 -
pytorch训练时报错NotImplementedError
根据报错信息“in _forward_unimplemented”，就知道是forward这个方法没实现好。有些人的错误可能是没实现forward或者forward拼写错误，我检查了好几次才发现我是缩进出了问题。大晚上的，确实头昏眼花，不熬了，水一篇帖子睡觉去了。敲代码时没检查，python的自动缩进在这里是错的。原创 2024-02-19 02:03:58 · 432 阅读 · 1 评论