0

我试图在计算引擎和谷歌云 TPU 上运行自定义估算器，但是我收到一个错误：

ValueError: shape [2] cannot be sharded 8 ways along dimension 0

我不知道是什么原因或如何解决它 - 任何想法？

安迪·P。

4

1 回答 1

0

看起来你正在使用一个二值向量。理想情况下，TPU 的所有尺寸都可以被 8 整除。如果你只是在玩，我建议只制作一个 8 长的向量。

更多信息：https ://cloud.google.com/tpu/docs/tpus#dimensions

于 2019-06-10T23:55:46.797 回答