TF-Keras mixed precision training leads to autograph errors about tf-keras

[tilakrayal]

I was able to reproduce the issue on tensorflow v2.16, tf-nightly whereas with the tensorflow v2.15 the same code was executed without any warnings.

WARNING: AutoGraph could not transform <function create_autocast_variable at 0x79fe61a00310> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: <gast.gast.Expr object at 0x79fe0b9ac850>

Kindly find the gist of it here. Thank you!

[hertschuh]

@lgeiger ,

Thanks for the bug report.

This issue was introduced when making Keras 3 the default in Tensorflow in version 2.16. The fix should be in the nightly build tomorrow.

Fabien

[lgeiger]

Thanks for the fix! Looks like the latest nightly indeed fixes this issue 👍

[tilakrayal]

@lgeiger,
As mentioned the TF-keras mixed precision issue was resolved and the warning is also disabled with the merged PR in the nightly.
KIndly find the gist for the reference. Thank you!