У меня есть тренировочный график, который я пытаюсь запустить на устройстве ARM. По сути, я использую contrib / pi_examples / label_image, но с моей сетью вместо Inception. Моя сеть была обучена с отсева, что теперь вызывает у меня проблемы:

Invalid argument: No OpKernel was registered to support Op 'Switch' with these attrs.  Registered kernels:
  device='CPU'; T in [DT_FLOAT]
  device='CPU'; T in [DT_INT32]
  device='GPU'; T in [DT_STRING]
  device='GPU'; T in [DT_BOOL]
  device='GPU'; T in [DT_INT32]
  device='GPU'; T in [DT_FLOAT]

 [[Node: l_fc1_dropout/cond/Switch = Switch[T=DT_BOOL](is_training_pl, is_training_pl)]]

Одно решение, которое я вижу, состоит в том, чтобы создать такую ​​статическую библиотеку TF, которая включает соответствующую операцию. С другой стороны, возможно, было бы лучше исключить выпадающие операции из сети, чтобы сделать ее проще и быстрее. Есть способ сделать это?

Благодарю.