时间戳

首页 > 解决方案 > 在 Colab 在 TPU 上训练模型时出错

tensorflow - 在 Colab 在 TPU 上训练模型时出错

问题描述

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

标签： tensorflow

解决方案

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

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

推荐阅读