Ich habe ein trainiertes eingefrorenes Diagramm, das ich auf einem ARM-Gerät ausführen möchte. Grundsätzlich verwende ich contrib / pi_examples / label_image, aber mit meinem Netzwerk anstelle von Inception. Mein Netzwerk wurde mit Dropout trainiert, was mir jetzt Probleme bereitet:

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

Eine Lösung, die ich sehen kann, besteht darin, eine solche statische TF-Bibliothek zu erstellen, die die entsprechende Operation enthält. Andererseits ist es möglicherweise besser, die Dropout-Operationen aus dem Netzwerk zu entfernen, um sie einfacher und schneller zu machen. Gibt es eine Möglichkeit, das zu tun?

Vielen Dank