Я преобразовал модель CNN-LSTM из потока в тензорный поток.
Сначала я использовал pytorch, преобразовал его в модель ONNX, а затем в модель с тензорным потоком, которая будет использоваться в Android Studio. Я успешно загружаю модель, но после этого сталкиваюсь с ошибками.
Кто-нибудь сталкивался с той же проблемой и достиг ее решения?
Заранее спасибо тем, кто откликнулся!
Сообщение об ошибке:
java.lang.IllegalArgumentException: No OpKernel was registered to support Op 'GatherV2' used by {{node GatherV2}}with these attrs: [Taxis=DT_INT32, Tindices=DT_INT64, Tparams=DT_INT64]
Registered devices: [CPU]
Registered kernels:
device='CPU'; Tparams in [DT_BOOL]; Tindices in [DT_INT64]
device='CPU'; Tparams in [DT_BOOL]; Tindices in [DT_INT32]
device='CPU'; Tparams in [DT_FLOAT]; Tindices in [DT_INT64]
device='CPU'; Tparams in [DT_FLOAT]; Tindices in [DT_INT32]
device='CPU'; Tparams in [DT_INT32]; Tindices in [DT_INT64]
device='CPU'; Tparams in [DT_INT32]; Tindices in [DT_INT32]