Customizing Keras Training with Model Subclassing
Learn how to subclass tf.keras.Model to override train_step, allowing for custom training logic while retaining the benefits of the Keras compile and fit API.
Loading…
ReadMeFeed / Community knowledge
Real questions. Useful conversations. Find the people who know your stack.
Learn how to subclass tf.keras.Model to override train_step, allowing for custom training logic while retaining the benefits of the Keras compile and fit API.
keras