Open aMIT-feat-aMEI opened 4 years ago
我用了gan.py里的原始代码,train的函数基本也是做按照视频里的代码写的,训练数据不是图片,是我自己的数据,然后训练大概四个epoch之后出现了内存爆炸的问题,程序自动kill了。 我的tensorflow是1.14版本,只是参考了视频里的代码,这里想问内存爆炸应该是什么原因?怎么样能让程序自动释放内存呢?
很抱歉打扰老师,希望老师能帮我解答一下,谢谢老师!!
祝好
我用了gan.py里的原始代码,train的函数基本也是做按照视频里的代码写的,训练数据不是图片,是我自己的数据,然后训练大概四个epoch之后出现了内存爆炸的问题,程序自动kill了。 我的tensorflow是1.14版本,只是参考了视频里的代码,这里想问内存爆炸应该是什么原因?怎么样能让程序自动释放内存呢?
很抱歉打扰老师,希望老师能帮我解答一下,谢谢老师!!
祝好