python - 如何在 TensorFlow v2.0 中正确应用梯度

Question

我正在尝试重现一些以前基于 TensorFlow v1.x 的神经转移教程。但使用 TensorFlow v2.0 .
出于某种原因，我无法使用 Adam optimizer 应用渐变由于，我相信，一些数据格式。
源代码非常大，所以我试图只提供最相关的行，但如果有人询问完整代码，我会在此处添加。
这是优化器

    opt = tf.optimizers.Adam(learning_rate = 5, beta_1 = 0.99, epsilon = 1e-1)

当我尝试将梯度应用于初始变量时

    opt.apply_gradients(zip(grads, init_image))

我收到错误

!opt.apply_gradients(zip(grads, init_image))
*** TypeError: zip argument #2 must support iteration

这些是 zip 参数:

init_image
<tf.Variable 'Variable:0' shape=(1, 345, 512, 3) dtype=float32, numpy=
array([[[[...]]]], dtype=float32)>

grads
<tf.Tensor: shape=(1, 345, 512, 3), dtype=float32, numpy=
array([[[[...]]]], dtype=float32)>

如您所见，形状与应有的相同，值在 numpy 内是一些实际值，而不是 nans或类似的东西。
梯度计算如下:

    with tf.GradientTape() as tape: 
         ...computing all_loss...
    total_loss = all_loss[0]
    grads = tape.gradient(total_loss, init_image)

请提出任何建议。

Answer 1

这应该很容易解决——问题是您需要将梯度/变量打包在 zip 的列表中。上类。所以简单地使用

opt.apply_gradients(zip([grads], [init_image]))

添加一些解释——您的代码可能会在 Eager 模式下运行而不会崩溃，但它可能会产生不正确的结果(必须对此进行测试)。使用 tf.function 时肯定会崩溃(从您提供的代码中不清楚您是否使用它)因为张量在图形模式下不可迭代。通过将它们包装在列表中，zip 现在可以遍历它们(即使只有一个元素)。