继续每天的Pytorch炼丹之路,今天又碰到了一个很奇葩的报错。错误提示信息如下图所示,可见多个Frame #***,每个Frame后接不同或类似的信息。
看到上图所示的错误提示,有点不知所云。继续往上翻,在Frame #0上方终于看到了报错的真正原因RuntimeError:CUDA out of memory,如下图所示。继续往上看,可以看到具体出错的是小编添加的pool模块。仔细查看代码,是该模块一个参数写错导致了占用显存过大。
继续每天的Pytorch炼丹之路,今天又碰到了一个很奇葩的报错。错误提示信息如下图所示,可见多个Frame #***,每个Frame后接不同或类似的信息。
看到上图所示的错误提示,有点不知所云。继续往上翻,在Frame #0上方终于看到了报错的真正原因RuntimeError:CUDA out of memory,如下图所示。继续往上看,可以看到具体出错的是小编添加的pool模块。仔细查看代码,是该模块一个参数写错导致了占用显存过大。