Google Colab需要太长时间来训练GAN

时间:2020-02-03 17:47:33

标签: python-3.x tensorflow machine-learning google-colaboratory generative-adversarial-network

我遵循了TensorFlow网站(下面链接)上的Generative-Adversarial Network教程,它说:“使用Colab的默认设置,这可能需要大约一分钟/一个纪元。”到目前为止,仅完成了两个纪元,每个纪元约为750秒。这意味着将需要10个小时来完成所有50个时期。我先尝试使用默认设置,然后尝试选择TPU,如果有的话,现在情况更糟。我希望发现任何错误,可以帮助加快GAN的培训速度。

代码太长,无法在此处放置,因此我在下面将指向本演练的链接放在下面。

链接到本教程:https://www.tensorflow.org/tutorials/generative/dcgan

链接到我的Google Colab代码:https://drive.google.com/open?id=10-VTSyFqWMRT3NIWOU77he7li4YRE0dY

1 个答案:

答案 0 :(得分:0)

您是否尝试过在“运行时”菜单中将“运行时类型”调整为GPU?