![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
神经网络
柠檬也可以不酸
我很懒,没有添加简介
展开
-
RuntimeError: CUDA out of memory.不一定是显存不够用
RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 23.70 GiB total capacity; 7.44 GiB already allocated; 87.88 MiB free; 7.71 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_.转载 2022-01-22 21:36:17 · 10027 阅读 · 5 评论 -
RuntimeError:——Expected dimensional不匹配
**RuntimeError: Expected 4-dimensional input for 4-dimensional weight [10, 1, 5, 5], but got 3-dimensional input of size [1, 28, 28] instead**网络的第一层此时我处理的是一张图片,而卷积神经网络模型是输入的是一个数据集的图片,而不是一张图片。此时我的输入维度如下:可通过如下方式来增加输入图片的维度:增加维度过后:至此,问题解决。附注:最开始我原创 2021-05-29 12:58:44 · 9224 阅读 · 1 评论