Resource ExhaustedError(有关追溯的回音):用形状分配张量[33694,33694]并键入floa



我正在尝试运行培训文件,并且此错误弹出。我的数据尺寸为4190 x 33694。

您正在尝试分配4.2GB的张量(= 33694*33694*4/1024^3(。在反向版期间,将分配相同大小的另一个张量。即使您的网络只有一个(完全连接?(层,您可能需要12GB的视频内存来运行它。

您将不得不查看您的设计,几乎没有任何方式。用不同的东西替换FC层,大大减少神经元的数量,如果可能

最新更新