使用Flux时，数据量过大总会OutOfMemery，有没有办法限制内存（显存）使用量以使程序正常进行
没看懂，
把batch_size调低点吧？
另外，理论上讲，前向计算的时候，可以一步一步地计算，把显存的占用降到尽可能地底，但一般人不会这么做…
备案号：京ICP备17009874号-2