Skip to content

Conversation

@abheesht17
Copy link
Collaborator

@abheesht17 abheesht17 commented Apr 21, 2025

To be merged after #46

@abheesht17 abheesht17 requested a review from hertschuh April 21, 2025 17:17
@abheesht17 abheesht17 mentioned this pull request Apr 21, 2025

@keras.saving.register_keras_serializable() # type: ignore[misc]
def default_rank_discount_fn(rank: types.Tensor) -> types.Tensor:
return ops.divide(ops.cast(1, dtype=rank.dtype), ops.log2(1 + rank))
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nitpick: ops.add(rank, 1.0)

@abheesht17 abheesht17 requested a review from hertschuh April 21, 2025 23:57
@hertschuh hertschuh merged commit 7f3ff33 into keras-team:main Apr 22, 2025
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants