标签: tensorflow
我正在尝试运行我的训练文件,并且弹出此错误。我的数据尺寸是33190的4190。我试图减小批处理的大小,但仍然没用
答案 0 :(得分:0)
您正在尝试分配大小为4.2GB(= 33694 * 33694 * 4/1024 ^ 3)的张量。反向传播期间将分配另一个相同大小的张量。即使您的网络只有一个(完全连接?)层,您可能也需要12GB的视频内存来运行它。
您将不得不审查您的设计,几乎没有解决办法。用不同的东西代替FC层,大幅减少神经元数量,并尽可能调整日期范围。