kubeflow / training-operator

Distributed ML Training and Fine-Tuning on Kubernetes
https://www.kubeflow.org/docs/components/training
Apache License 2.0
1.61k stars 700 forks source link

Update TF examples to Keras V3 #2319

Open YosiElias opened 2 weeks ago

YosiElias commented 2 weeks ago

What happened?

We updated the TF examples to use TF-V2 (#2015), but as described here, as a workaround it uses Keras V2.

We need to verify whether the code can be adapted to work with Keras V3.

What did you expect to happen?

TensorFlow examples are executed using Keras V3

Impacted by this bug?

Give it a 👍 We prioritize the issues with most 👍

tenzen-y commented 2 weeks ago

/remove-label lifecycle/needs-triage