在TPU上的Colab中训练模型时出错

Question

我正在尝试使用TensorFlow构建GAN，并希望在TPU上对其进行培训。但是我得到下面提到的错误 '确保插槽变量是在相同策略范围内创建的。如果您要从范围以外的检查点还原，则可能会发生这种情况。 我是TensorFlow的新手。我所有的变量都是使用tf.distribute.experimental.TPUStrategy

创建的

Answer 1

您使用的是带有自定义层的模型或最新版本的 TensorFlow。如果是这样，则必须在正确的范围内调用 load_weights。像这样：

strategy = TPUStrategy(resolver)
with strategy.scope():
    model = ...
    model.compile(...)
    model.load_weights(...)