Limpe as operações de remoção do gráfico TensorFlow
Eu tenho um gráfico congelado treinado que estou tentando executar em um dispositivo ARM. Basicamente, estou usando contrib / pi_examples / label_image, mas com minha rede em vez de Inception. Minha rede foi treinada com abandono, o que agora me causa problemas:
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)]]
Uma solução que posso ver é criar uma biblioteca estática TF que inclua a operação correspondente. Por outro lado, pode ser uma idéia melhor eliminar as operações de abandono da rede para torná-la mais simples e rápida. Existe uma maneira de fazer isso?
Obrigado.