You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thanks for reporting the issue. I have tested the code snippet and reproduces the reported behaviour. Attached gist file for reference.
We will look into the issue and update you the same.
Current Behavior:
On Keras 3 with TensorFlow 2.16.1,
DropBlock2D
fails on graph mode with the following casting error:Steps To Reproduce:
See the following collab:
DropBlock2D Graph Mode
Version:
Anything Else:
The fix is trivial, just use
ops.cast()
instead of pythonint()
in thecall
method of theLayer
.The text was updated successfully, but these errors were encountered: