2017-02-04 2 views
0

Включен ли FIFOQueue в TensorFlow для iOS?Поддерживается ли FIFOQueue в TensorFlow для iOS?

Я тренировался модель, заморозил его, и оптимизирован для вывода, но я получаю сообщение об ошибке FIFOQueue на прошивке

«Не удался создать TensorFlow графика: недопустимый аргумент: Нет OpKernel был зарегистрирован для поддержки Op Зарегистрированные устройства: [CPU], зарегистрированные ядра: [[Узел: fifo_queue = FIFOQueuecapacity = 30, component_types = [DT_FLOAT, DT_FLOAT, DT_FLOAT], container = "", forms = [[640,480,3 ], [300,1,2], [300,1,4]], shared_name = ""]] "

Я даже попробовал установку is_training = False в tensorflow/contrib/slim/python/slim/nets/inception_v1.py перед обучением и получением моего ввода_graph.pb. (Я пробовал контрольную точку, созданную с помощью is_training = True)

Является ли FIFOQueue не поддерживаемым в TensorFlow для iOS?

+0

Возможно, это не проблема с iOS. Это проблема с размещением устройства (что делает жесткий запрос для процессора только для работы на GPU) или проблемой dtype (float/double). Я бы предположил, что первый, и в этом случае возможно [allow_soft_placement] (https://www.tensorflow.org/how_tos/using_gpu/) поможет? –

ответ

0

Я добавил «tensorflow/core/kernels/fifo_queue_op.cc» в файл tensorflow/contrib/makefile/tf_op_files.txt, затем запустил compile_ios_tensorflow.sh, и ошибки исчезли.

Смежные вопросы