25得票6回答
在Tensorflow 2.0中自定义训练循环中应用回调函数

我正在使用Tensorflow DCGAN实现指南中提供的代码编写自定义训练循环。我想在训练循环中添加回调函数。在Keras中,我知道我们将它们作为参数传递给“fit”方法,但找不到有关如何在自定义训练循环中使用这些回调函数的资源。我正在添加来自Tensorflow文档的自定义训练循环的代码:...

7得票1回答
使用tf.function时如何计算梯度

我对以下示例中的行为感到困惑: import tensorflow as tf @tf.function def f(a): c = a * 2 b = tf.reduce_sum(c ** 2 + 2 * c) return b, c def fplain(a)...