Open Mrutyunjay01 opened 12 months ago
The dtype string
is only supported with the TF backend. So there's another issue here, which is that the error message is not clear. It should just say that dtype string isn't supported with the JAX backend.
@Mrutyunjay01, Could you please provide sample reproducible code, the tutorial you have mentioned has other error to solve before we get to the reported issue part.
Apologies for late response. I faced the issue when tried to port BERT MLM to backend agnostic. Been a while, let me refine the port code once, and see if the issue persists still.
cc: @sachinprasadhs
Update:
The issue no longer persists in keras 3.3. So, we can close this. Currently I am trying to port BERT MLM to backend agnostic, will report if I come across any such issues.
Reopening the issue as faced in the draft PR mentioned.
As mentioned keras-team/keras-core#843, I was trying further modify the mlm-bert pipeline for keras-core with backend-agnostic support. The last part of the example tries to create an end-to-end pipeline with raw texts as the input to the model. Mentioned as follows:
But while executing with
jax
backend, it throws the following error:Is having the input layer of dtype `string' might cause the problem here?