TensorFlow 2.4 发布

栏目: 软件资讯 · 发布时间: 5年前

内容简介:TensorFlow 2.4 发布。主要更新包括对分布式训练和混合精度的支持,新的 NumPy 前端以及用于监视和诊断瓶颈的工具,此外还涉及性能和扩展方面的增强。 tf.distribute 中的新功能 参数服务器策略 在 2.4 中,tf.di...

TensorFlow 2.4 发布。主要更新包括对分布式训练和混合精度的支持,新的 NumPy 前端以及用于监视和诊断瓶颈的工具,此外还涉及性能和扩展方面的增强。

tf.distribute 中的新功能

  • 参数服务器策略

在 2.4 中,tf.distribute模块引入了使用 ParameterServerStrategy 对模型进行异步训练的实验支持。以及自定义训练循环。参数服务器训练集群由工作服务器和参数服务器组成。变量在参数服务器上创建,然后在每个步骤中由工作人员读取和更新。变量的读取和更新在整个工作进程中独立发生,而没有任何同步。由于工作进程彼此不依赖,因此该策略具有工作程序容错的优势,并且在使用可抢占式 VM 时很有用。

  • Multi Worker Mirrored Strategy

MultiWorkerMirroredStrategy 现在不再处于实验阶段,已经成为稳定 API 的一部分。像它的同类产品MirroredStrategy一样,MultiWorkerMirroredStrategy 通过同步数据并行实现分布式训练,可跨多台机器。

TensorFlow 2.4 发布

Keras 更新

  • 混合精度

Keras 混合精度 API 也已脱离实验性,成为稳定的 API。大多数 TensorFlow 模型使用 float32 dtype; 但是,有些低精度类型(例如 float16)使用的内存更少。混合精度是在同一模型中使用 16 位和 32 位浮点类型以进行更快的训练。该 API 可以将模型性能在 GPU 上提高 3 倍,在 TPU 上提高 60%。

  • 优化器

此版本重构tf.keras.optimizers.Optimizer 类,使model.fit或定制训练循环的用户能够编写可与任何优化程序一起使用的训练代码。

其他

  • tf.experimental.numpy

TensorFlow 2.4 引入了对 NumPy API 子集的实验性支持,该模块使用户可以运行由 TensorFlow 加速的 NumPy 代码。

  • 新的探查工具

TensorFlow Profiler 是可以用来衡量 TensorFlow 模型的训练表现和资源消耗的 工具 套件,可用来了解模型中操作的硬件资源消耗,诊断瓶颈并进行更快地训练。

更新详情查看:https://blog.tensorflow.org/2020/12/whats-new-in-tensorflow-24.html


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

测试驱动开发

测试驱动开发

Kent Beck / 孙平平、张小龙 / 中国电力出版社 / 2004-4-1 / 28.00元

《测试驱动开发》(中文版)设想把编程看成是转动曲柄从井里提一桶水上来的过程。如果水桶比较小,那么仅需一个能自由转动的曲柄就可以了。如果水桶比较大而且装满水,那么还没等水桶全部被提上来你就会很累了。你需要一个防倒转的装置,以保证每转一次可以休息一会儿。水桶越重,防倒转的棘齿相距越近。测试驱动开发中的测试程序就是防倒转装置上的棘齿。一旦我们的某个测试程序能工作了,你就知道,它从现在开始并且以后永远都可......一起来看看 《测试驱动开发》 这本书的介绍吧!

JS 压缩/解压工具
JS 压缩/解压工具

在线压缩/解压 JS 代码

URL 编码/解码
URL 编码/解码

URL 编码/解码

MD5 加密
MD5 加密

MD5 加密工具