有替代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 import trainer_lib

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

Answer 2

看来你混合了不同版本的Tensor2Tensor。在最近的版本中，没有tpu_trainer_lib（已重命名）。即使在存在此类文件的少数版本中，它也支持TPU，GPU和CPU的培训，对于TPU特定的代码，只有一些if-then-else分支。 顺便提一下，TPU目前仅适用于Google员工。