google-colaboratory - 运行时错误 :Mixing different tf. 分发。策略对象

Question

您好!我在使用TPU编译模型时遇到了一些问题，部分代码如下:

resolver = tf.contrib.cluster_resolver.TPUClusterResolver(TF_MASTER)

tf.contrib.distribute.initialize_tpu_system(resolver)

strategy = tf.contrib.distribute.TPUStrategy(resolver)

with strategy.scope():

  model = create_model()

  model.compile(optimizer=tf.keras.optimizers.Adadelta(),loss='categorical_crossentropy',metrics='accuracy'])

我得到了运行时错误: enter image description here

你能帮助我吗？

Answer 1

我通过各种困惑尝试解决了我的问题。您可以重新启动程序或注释代码:

resolver = tf.contrib.cluster_resolver.TPUClusterResolver
tf.contrib.distribute.initialize_tpu_system(resolver)
strategy = tf.contrib.distribute.TPUStrategy(resolver)

with strategy.scope():
  model = create_model()
  model.compile()

为避免问题