From 8423d3c1a0fb34919f933909a1c4759b3134685d Mon Sep 17 00:00:00 2001 From: David Ryglicki Date: Mon, 14 Oct 2024 14:29:25 -0400 Subject: [PATCH] Adding stop_gradient back since it is now available in ops. --- keras/src/layers/normalization/spectral_normalization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras/src/layers/normalization/spectral_normalization.py b/keras/src/layers/normalization/spectral_normalization.py index 727d6bb58dbd..fc11844fc929 100644 --- a/keras/src/layers/normalization/spectral_normalization.py +++ b/keras/src/layers/normalization/spectral_normalization.py @@ -105,8 +105,8 @@ def normalized_weights(self): ops.matmul(vector_u, ops.transpose(weights)), axis=None ) vector_u = normalize(ops.matmul(vector_v, weights), axis=None) - # vector_u = tf.stop_gradient(vector_u) - # vector_v = tf.stop_gradient(vector_v) + vector_u = ops.stop_gradient(vector_u) + vector_v = ops.stop_gradient(vector_v) sigma = ops.matmul( ops.matmul(vector_v, weights), ops.transpose(vector_u) )