Skip to content

Commit 017d350

Browse files
committed
Missed optimizer
1 parent 0e28efe commit 017d350

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_addons/image/dense_image_warp_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def test_gradients_exist(self):
204204
interp = dense_image_warp(image, flows)
205205
loss = tf.math.reduce_mean(tf.math.square(interp - image))
206206

207-
optimizer = tf.optimizers.Adam(1.0)
207+
optimizer = tf.keras.optimizers.Adam(1.0)
208208
grad = tf.gradients(loss, [flows])
209209
opt_func = optimizer.apply_gradients(zip(grad, [flows]))
210210
init_op = tf.compat.v1.global_variables_initializer()

0 commit comments

Comments
 (0)