tf.function功能提升绘图性能

在TensorFlow 2中,默认情况下将打开急切执行功能。用户界面直观而灵活(运行一次性操作要容易得多且更快),但这可能会牺牲性能和可部署性。

您可以使用tf.function程序制作图形。它是一种转换工具,可以从您的Python代码中创建与Python无关的数据流图。这将帮助您创建高性能和可移植的模型,并且需要使用SavedModel

本指南将帮助您概念化如何tf.function在引擎盖下工作,以便您可以有效地使用它。

主要要点和建议是:

  • 在紧急模式下进行调试,然后使用进行修饰@tf.function
  • 不要依赖于Python的副作用,例如对象突变或列表追加。
  • tf.function与TensorFlow ops配合使用效果最佳;

你可能感兴趣的:(Python,python)