Как преобразовать модель tenorflow seq2seq в тензор потока lite? - PullRequest
0 голосов
/ 16 апреля 2020

Я использую пример модели seq2seq с сайта tenorflow. https://www.tensorflow.org/tutorials/text/nmt_with_attention Я обучил его на своих собственных данных и хочу запустить его на Raspberry Pi. Сначала я попытался скопировать код и загрузить контрольные точки, но это не сработало. Затем я попытался преобразовать свою модель в tflite, но я не знаю, как это сделать правильно, и не могу найти, как это сделать. Есть 3 модели для сохранения, кодировщик, декодер и внимание. Код encoder.save("encoder") после загрузки контрольных точек выдает ошибку

ValueError: Model <__main__.Encoder object at 0x000002F72CEC7D08> cannot be saved because the input shapes have not been set. Usually, input shapes are automatically determined from calling .fit() or .predict(). To manually set the shapes, call model._set_inputs(inputs).

В этом примере не используются fit () или Forex (), так как его сохранить и преобразовать в tflite?

...