Сконфигурируйте Tensorflow для CPU с помощью Java API

Question

Я пытаюсь играть с числами для intra_op_parallelism_threads inter_op_parallelism_threads .Я использую Tensorflow и Proto версии 1.8.0 для 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();

Когда я запускаю вывод, я получаю следующую ошибку: Тайм-аут ожидания уведомления.

Я делаю это неправильно?Есть ли другой подход?

Спасибо

Answer 1

У меня та же проблема, и вот мой выход.

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