tf.function功能提升绘图性能

发布于:2021-05-18 10:45:00

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


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


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


主要要点和建议是:



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

相关推荐

最新更新

猜你喜欢