Python 有没有tpu_trainer_lib的替代方案？

我一直在尝试在本地运行google的tensor2tensor（），但当它达到这一点时，它就卡住了：

from tensor2tensor.tpu import tpu_trainer_lib

这是有道理的，因为我的Mac中没有TPU。有没有其他替代功能

这是错误消息

ModuleNotFoundError: No module named 'tensor2tensor.tpu'

---

您应该能够将其替换为：

从tensor2tensor.utils导入培训师库

---

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

您应该能够将其替换为：

从tensor2tensor.utils导入培训师库

---

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

看来您混合了不同版本的Tensor2Tensor。在最新版本中，没有

tpu\u trainer\u lib

（已重命名）。即使在这种文件存在的少数版本中，它也支持TPU、GPU和CPU上的培训，只有一些针对TPU特定代码的if-then-else分支。

---

顺便说一句，TPU目前仅适用于谷歌员工。

您似乎混合了不同版本的Tensor2Tensor。在最新版本中，没有

tpu\u trainer\u lib

（已重命名）。即使在这种文件存在的少数版本中，它也支持TPU、GPU和CPU上的培训，只有一些针对TPU特定代码的if-then-else分支。 顺便说一句，TPU目前仅适用于谷歌员工