tpu_trainer_lib 有替代方案吗？

Question

我一直在尝试 运行 google 的 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 目前仅供 Google 名员工使用。