Recommended techniques to employ for idiomatic TF 2.0

The first recommendation concerns dealing with a general code workflow in TF 2.0. A common workflow in TF 1.x was to use a waterfall strategy, where all of the computations were laid out onto the default graph. Then, selected tensors were run using In TF 2.0, code should be refactored into smaller functions that will be called as needed. These functions can be normal Python functions and can still be run in graph mode if they're called inside another function annotated with tf.function. This means that tf.function should only be used to annotate high-level computations, such as the forward pass of a model or a single training step.

Previously, all of the computations that ...

Get What's New in TensorFlow 2.0 now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.