“预期的`model`参数是一个Model实例,得到了keras对象

时间:2019-06-01 17:28:53

标签: keras

我正在尝试在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>

感谢您的帮助。谢谢!

1 个答案:

答案 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