我正在尝试在colab上使用TPU,
model.compile(optimizer='adam',
loss='binary_crossentropy',
metrics=[auc])
model = tf.contrib.tpu.keras_to_tpu_model(
model,
strategy=tf.contrib.tpu.TPUDistributionStrategy(
tf.contrib.cluster_resolver.TPUClusterResolver(TPU_ADDRESS)))
我收到以下错误。
预期model
参数是一个Model
实例,得到了<keras.engine.training.Model object at 0x7f8e828822e8>
感谢您的帮助。谢谢!
答案 0 :(得分:0)
也许首先尝试将未编译的模型转换为TPU模型,然后再编译TPU模型。
就像我在这篇中等文章中找到的那样。 https://medium.com/the-artificial-impostor/keras-for-tpus-on-google-colaboratory-free-7c00961fed69
这是Google colab链接https://colab.research.google.com/gist/ceshine/f196d6b030adb1ec3a8d0b50642709dc/keras-fashion-mnist-tpu.ipynb#scrollTo=pWEYmd_hIWg8