标签: python python-3.x tensorflow bazel
我正在尝试将smartreply.tflite模型加载到python解释器中,因此我可以用它来基于问题生成答复。该模型使用3个自定义操作(predict,normalize和extract_features)。
smartreply.tflite
predict
normalize
extract_features
我能够成功地将三个操作加载到解释器中,但是,当我尝试将模型加载到解释器中时,它给我一个错误,告诉我找不到操作Normalize。
Normalize
我希望在加载tflite模型时不会出错。
我怀疑当我尝试加载模型时,它只会在内置目录中查找所需的操作。