有没有tpu_trainer_lib的替代品？

Question

我一直在尝试在本地运行谷歌的tensor2tensor (https://github.com/tensorflow/tensor2tensor)，但当它运行到下面这一步时，它就卡住了：

from tensor2tensor.tpu import tpu_trainer_lib

这就说得通了，因为我的Mac里没有TPU。有没有替代该功能的方法？

下面是错误消息

ModuleNotFoundError: No module named 'tensor2tensor.tpu'

Answer 1

您应该能够将其替换为：

从tensor2tensor.utils导入trainer_lib

然后从trainer_lib调用适当的方法。

Answer 2

看起来你混合了不同版本的Tensor2Tensor。在最近的版本中，没有tpu_trainer_lib (已重命名)。即使在少数存在这样的文件的版本中，它也支持TPU、GPU和CPU上的训练，只有一些针对TPU特定代码的if-then-else分支。顺便说一句，TPU目前只对谷歌员工可用。