Open lseongjoo opened 3 weeks ago
Hi for the API's or Layers which does not support XLA
then the model will not proceed with jit_compile=True
, it will instead fallback to jit_compile=False
mode.
This issue is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you.
Description
When training a model in Keras 3.3 with the
jit_compile=True
option, a warning is displayed:PyTorch reports back its backends
Investigation
Upon reviewing the source code, it appears that the
jit_compile
option is not implemented for PyTorch models. This feature might be planned but is not yet reflected in the code.Steps to Reproduce
jit_compile=True
.Expected Behavior
The model should utilize
jit_compile=True
without warnings, or the documentation should clarify that this option is not supported for PyTorch models.Environment
Additional Information
Please provide any updates on the planned implementation of
jit_compile
for PyTorch models or correct the documentation if this feature is not intended to be supported.