Tensorflow Serving分组卷积

时间:2018-09-18 04:04:17

标签: python tensorflow keras tensorflow-serving

我正在使用keras和tensorflow进行笔迹检测。我已经准备好模型并使用hdf5文件。当我尝试使用 tensorflow服务提供服务时,出现以下错误:

grpc.framework.interfaces.face.face.LocalError: LocalError(code=StatusCode.UNIMPLEMENTED, details="Generic conv implementation does not support grouped convolutions for now.
 [[{{node conv2d_1/convolution}} = Conv2D[T=DT_FLOAT, _output_shapes=[[?,40,40,20]], data_format="NHWC", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_conv2d_1_input_0_0, conv2d_1/kernel/read)]]")

我使用了this文章中提到的两个文件来从我的hdf5文件(正在运行)生成模型。请提供任何可帮助我解决此问题的意见?

1 个答案:

答案 0 :(得分:1)

我遇到了类似的错误,但错误消息为“融合的conv实现不支持...”,而不是“通用的conv实现不支持...”。事实证明,这也是我的TensorFlow版本问题-我需要的是1.13.2版本。为了解决这个问题,我在做“将tensorflow导入为tf”之前做了此操作:

 !pip install tensorflow==1.13.2

这为我解决了。