Custom loss usage by name: not working in V3 but worked in V2

[shkarupa-alex]

Default use case for custom loss in V2 (for me) was:

1. Register custom loss with register_keras_serializable
2. Use it in model compile by name compile(loss="MyPackage>MyLoss")

This worked in Keras V2, but not working in Keras V3. Here is a simple test https://colab.research.google.com/drive/1R3z3fFhV7NN1Oa_HJYp60LsneoxjOlvj?usp=sharing

[chunduriv]

@shkarupa-alex,

Unfortunately, I don't have access to Colab. Could you please grant us access?

[shkarupa-alex]

shared

[nkovela1]

Hi @shkarupa-alex, thanks for the Colab! Could you try keras.saving.get_registered_object(name) with the name "MyPackage>MyLoss"?

[shkarupa-alex]

Hi @shkarupa-alex, thanks for the Colab! Could you try keras.saving.get_registered_object(name) with the name "MyPackage>MyLoss"?

Added to the same colab notebook: keras.saving.get_registered_object("MyPackage>MyLoss") returns custom loss class