python - 有没有 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 上进行训练，只有一些 if-then-else 分支用于 TPU 特定代码。顺便说一句，TPU 目前仅适用于 Google 员工。