使用Java API为CPU配置Tensorflow

Question

我正在尝试使用内部线程和内部线程的数字。我使用Tensorflow和Proto version 1.8.0 for Java。 我使用了以下代码：

     ConfigProto config = ConfigProto.newBuilder()
                            .setInterOpParallelismThreads(44)
                            .setIntraOpParallelismThreads(44).build();

     outputs = model.session().runner().setOptions(config.toByteArray()).
                     feed("image_tensor", input).fetch("detection_scores").fetch("detection_classes").fetch("detection_boxes").fetch("num_detections").run();

当我运行a推断时，我得到以下错误：等待通知超时。

我做错了吗？还有别的办法吗

谢谢!

Answer 1

我也有同样的问题，这是我的出路

SavedModelBundle.loader("model_dir").withTags("tag").withConfigProto(config.toByteArray()).withXX().load();