我正在尝试运行培训文件,并且此错误弹出。我的数据尺寸为4190 x 33694。
您正在尝试分配4.2GB的张量(= 33694*33694*4/1024^3(。在反向版期间,将分配相同大小的另一个张量。即使您的网络只有一个(完全连接?(层,您可能需要12GB的视频内存来运行它。
您将不得不查看您的设计,几乎没有任何方式。用不同的东西替换FC层,大大减少神经元的数量,如果可能
我正在尝试运行培训文件,并且此错误弹出。我的数据尺寸为4190 x 33694。
您正在尝试分配4.2GB的张量(= 33694*33694*4/1024^3(。在反向版期间,将分配相同大小的另一个张量。即使您的网络只有一个(完全连接?(层,您可能需要12GB的视频内存来运行它。
您将不得不查看您的设计,几乎没有任何方式。用不同的东西替换FC层,大大减少神经元的数量,如果可能
javascript python java c# php android html jquery c++ css ios sql mysql arrays asp.net json python-3.x ruby-on-rails .net sql-server django objective-c excel regex ruby linux ajax iphone xml vba spring asp.net-mvc database wordpress string postgresql wpf windows xcode bash git oracle list vb.net multithreading eclipse algorithm macos powershell visual-studio image forms numpy scala function api selenium