将具有dtype float64的图形转换为float32

时间:2019-07-29 11:49:25

标签: python-3.x tensorflow

我有一个图,但所有张量类型均为float64。为了利用GPU加速,必须使用float32。我唯一拥有的是一个检查点文件。我可以将其从float64转换为float32吗?

我认为可以通过解析图结构并手动重建图来完成。但这花费了很多时间。我想知道是否有为此建议的工具。

another question的问题不同。在他们的情况下,他们具有用于图形构建的代码。就我而言,我没有那个(除非我手动编写)。

0 个答案:

没有答案