New content on http://keras.io : Keras debugging tips.https://keras.io/examples/keras_recipes/debugging_tips/ …
Show this thread
-
I always wonder when/where to use the
@tf.function decorator. In your example provided here, would it be useful to use it to decorate the train_step function ?
If you're using fit/predict, you never need to use tf.function (just pass run_eagerly=True/False in compile). If you're writing low-level training loops from scratch, add a tf.function decorator to your training step (just one in total).
