ValueError：操作u'tpu_140462710602256 / VarIsInitializedOp'已标记为不可提取

Question

代码在GPU和CPU上正常工作，但是当我使用keras_to_tpu_model函数使模型能够在TPU上运行时，发生了错误。

这是colab的完整输出：https://colab.research.google.com/gist/WangHexie/2252beb26f16354cb6e9ba2639970e5b/tpu-error.ipynb
将运行类型更改为TPU，我认为可以重现。

代码在github上：https://github.com/WangHexie/DHNE/blob/master/src/hypergraph_embedding.py#L60

您可以通过更改为gpu分支来在GPU上测试代码。

追踪

if (!App.selectedCardCount.HasValue)
   App.selectedCardCount = App.DB.GetSelectedCardCount();

vm.x = App.selectedCardCount.Value.ToString("N0") + " x ";

Answer 1

我有一个同样的问题，使我困惑了两天。我发现一种解决方案是仅切换为使用tf.train.RMSPropOptimizer而不是使用tensorflow.keras.optimizers中的RMSProp。