Tensorflow内置训练循环与自定义训练循环的性能

时间:2020-04-17 20:34:45

标签: tensorflow2.0

我刚刚使用tf.keras.Model在tensorflow 2.0中编码了一个子类化模型,并且我使用了一个累加的训练循环。培训是永远的!我遇到了这个tensorflow示例VAE,它与我的模型以及如何自定义训练循环非常相似。我已经使用定制和内置的训练功能在链接中运行了tensorflow示例,前者大约需要10秒的90秒,而后者只需要20秒。

为什么自定义循环速度较慢,并且如何改进以使其与内置拟合功能一样快?这是一个仅在run_vae

上方的链接中运行VAE的链接。
problem for my own model in tensorflow example 

0 个答案:

没有答案