谷歌TensorFlow 2.2.0正式发布:道别Python 2,拥抱Python 3

谷歌

 TensorFlow 2.2.0 正式发布了,该版本终止了对 Python 2 的支持。现在,新的 TensorFlow Docker 镜像版本仅提供 Python 3。

image.png

主要特性和改进

· 将字符串张量的标量类型从 std::string 替换为 tensorflow::tstring

· TF 2 的新 Profiler,用于 CPU/GPU/TPU。它提供设备和主机性能分析,包括输入管道和 TF Ops。

· 不推荐使用 SWIG,而是使用 pybind11 将 C++ 函数导出到 Python,这是弃用 Swig 所作努力的一部分。

· tf.distribute:

通过使用新添加的 tf.keras.layers.experimental.SyncBatchNormalization 层,添加了对全局同步 BatchNormalization 的支持。该层将在参与同步训练的所有副本之间同步 BatchNormalization 统计信息。

使用 tf.distribute.experimental.MultiWorkerMirroredStrategy 提高 GPU 多工分布式培训的性能

· tf.keras:

Model.fit  的主要改进:

现在,SavedModel 格式支持所有 Keras 内置层(包括指标,预处理层和有状态 RNN 层)。

更新 Keras 批处理规范化层,以使用 fused_batch_norm 中的运行平均值和平均值计算。

· tf.lite:

默认情况下启用 TFLite 实验性新转换器。

· XLA

XLA 现在可以在 Windows 上构建并运行。所有预构建的软件包都随附有 XLA。

可以在 CPU 和 GPU 上使用“编译或抛出异常”语义为 tf.function 启用 XLA。

新版本包含大量 bug 修复等,详情可见更新说明:

https://github.com/tensorflow/tensorflow/releases/tag/v2.2.0

相关文章