We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2c3f2e6 commit bd63e54Copy full SHA for bd63e54
keras_rs/src/losses/list_mle_loss.py
@@ -208,7 +208,7 @@ def call(
208
losses = ops.multiply(losses, weights)
209
losses = ops.squeeze(losses, axis=-1)
210
return losses
211
-
+ # getting config
212
def get_config(self) -> dict[str, Any]:
213
config: dict[str, Any] = super().get_config()
214
config.update({"temperature": self.temperature})
0 commit comments