dragen1860 / Deep-Learning-with-TensorFlow-book

深度学习入门开源书,基于TensorFlow 2.0案例实战。Open source Deep Learning book, based on TensorFlow 2.0 framework.
http://www.ai101edu.com
13.2k stars 4.09k forks source link

第十三章gan训练的问题 #200

Open aMIT-feat-aMEI opened 4 years ago

aMIT-feat-aMEI commented 4 years ago

我用了gan.py里的原始代码,train的函数基本也是做按照视频里的代码写的,训练数据不是图片,是我自己的数据,然后训练大概四个epoch之后出现了内存爆炸的问题,程序自动kill了。 我的tensorflow是1.14版本,只是参考了视频里的代码,这里想问内存爆炸应该是什么原因?怎么样能让程序自动释放内存呢?

很抱歉打扰老师,希望老师能帮我解答一下,谢谢老师!!

祝好