使用tf估计器multi gpu进行手动混合精度训练

时间:2019-09-23 22:25:21

标签: tensorflow nvidia tensorflow-estimator

我想在多GPU上使用混合精度训练。在文档中,nvidia提到自动混合精度目前无法用于多GPU,因此,我浏览了有关手动混合精度训练的nvidia文档,它以低级tensorflow编写。他们没有给出清晰的解释,也没有很好的例子说明如何使用tf估计器对多GPU进行手动混合精度训练。我们可以在多GPU上使用手动混合精度吗?如果可以,我们可以将其与tf估算器一起使用吗?

0 个答案:

没有答案