首页 > 解决方案 > 在 colab 中加载 tensorflow lite edgetpu 模型

问题描述

我有一个为 edgetpu 编译的 tf lite 模型。

我想通过做类似的事情来恢复 tensorflow 中的模型权重所以 我不必再次训练

我没有edge-tpu,我想在colab中运行代码。当我尝试加载模型时

interpreter = tf.lite.Interpreter(
    'mymodel_edgetpu.tflite'
)
interpreter.allocate_tensors()

抛出以下错误:

遇到未解决的自定义操作:edgetpu-custom-op。节点号 1 (edgetpu-custom-op) 准备失败。

标签: tensorflowgoogle-colaboratoryedge-tpu

解决方案


推荐阅读