Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit 808cd5fd authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Fix 'build_keras_loss' for TensorFlow 2.13.

parent 02afef6e
No related branches found
No related tags found
1 merge request!50Fix compatibility issues with scikit-learn 1.3.0 and tensorflow 2.13.0
Pipeline #830863 passed
......@@ -96,7 +96,7 @@ def build_keras_loss(
loss = tf.keras.losses.deserialize(loss)
loss.reduction = reduction
# If this failed, try wrapping the function using LossFunction.
except ValueError:
except (AttributeError, ValueError):
loss = LossFunction(cls)
# Case when the string was deserialized into a class.
else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment