在Google Colab上,使相同代码(dcgan)收敛于tensorflow1.15但在tensorflow2.0中崩溃

我正在使用keras模块在tensorflow2.0.0和tensorflow1.15.0上运行dcgan代码,相同的代码在tensorflow1.15.0上收敛但在tensorflow2.0.0上崩溃

大约一个星期以来,我一直在尝试优化dcgan(深度卷积生成对抗网络),因此我决定从github上的代码中获得一些帮助,因此我从以下网址下载了dcgan mnist代码:

https://github.com/eriklindernoren/Keras-GAN/blob/master/dcgan/dcgan.py

和dcgan cifar10代码来自:

https://github.com/mafda/generative_adversarial_networks_101/blob/master/src/cifar10/02_DCGAN_CIFAR10.ipynb

令人惊讶的是,它们都没有在安装了tensorflow2.0.0的台式机上收敛,但是在使用tensorflow1.15.0时它们在google colab上收敛,而在tensorflow 2.0.0上尝试它们时再次崩溃了

这是tensorflow1.15代码的链接:

https://colab.research.google.com/drive/1L76-q5TrstH1qxcUXBHvacN3I9idIoAy

和tensorflow2.0代码:

https://colab.research.google.com/drive/1RZOM1hNyl9XoJQOznoDfIya80CS3QWCA

左图用于tensorflow1.15,右图用于tensorflow2.0:

cifar10:

https://drive.google.com/file/d/1EgzPvIn_hn-KwG99utAtybxJIiFVRqzs/view?usp=sharing

mnist10:

https://drive.google.com/file/d/1BnevcAO9xerDb6wsr0NP0hEkDdEHLm4z/view?usp=sharing

我已将两种代码的导入更改为以下代码段,因此我可以在tensorflow1.15和tensorflow2.0上均等地运行它们,而无需进行任何其他更改:

from tensorflow.keras.datasets import mnist
from tensorflow.keras.layers import Input,Dense,Reshape,flatten,Dropout,multiply
from tensorflow.keras.layers import BatchNormalization,activation,Embedding,ZeroPadding2D
from tensorflow.keras.layers import LeakyReLU
from tensorflow.keras.layers import UpSampling2D,Conv2D
from tensorflow.keras.models import Sequential,Model
from tensorflow.keras.optimizers import Adam

考虑到代码完全相同的事实,我不知道是什么原因会导致这种问题!!!

hejunyan2010 回答:在Google Colab上,使相同代码(dcgan)收敛于tensorflow1.15但在tensorflow2.0中崩溃

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3165868.html

大家都在问