Tensorflow Serving 分组卷积

5

我正在使用keras和tensorflow进行手写字体检测。我已经准备好了模型,并且使用hdf5文件运行良好。但是,当我尝试使用tensorflow serving提供服务时,出现以下错误:

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)]]")

我使用了这篇文章中提到的两个文件,从我的hdf5文件(已经可用)生成模型。请提供任何可能帮助我解决问题的输入?


构建模型块时,我遇到了类似的问题。然而,这是因为Tensorflow版本1.14。将其降级为1.13.1解决了这个问题。 - Narahari B M
1个回答

1
我遇到了类似的错误,但错误信息是“Fused conv implementation does not support...”,而不是“Generic conv implementation does not support...”。后来发现这也是因为我的TensorFlow版本问题 -- 我需要的是1.13.2版本。为了解决这个问题,在执行“import tensorflow as tf”之前,我做了以下操作:
 !pip install tensorflow==1.13.2

这对我解决了问题。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接