添加@tf.function装饰器

通常，官方发布或设计良好的脚本都会自带@tf.function装饰器或者允许您通过脚本入参控制，您无需额外添加。

特别地，使用Keras下Model.fit接口训练的脚本，TF2会在接口内部封装function，同样无需额外添加。

如果脚本中没有添加@tf.function装饰器，您可以先阅读TF2对@tf.function的使用说明，详情见链接，然后为您的训练/验证/推理函数添加@tf.function装饰器，并保证其在CPU或GPU环境下能正常工作。