Можем ли мы использовать TF-lite для переобучения? - PullRequest
Новая
       80

Можем ли мы использовать TF-lite для переобучения?

0 голосов
/ 19 июня 2020

Я преобразовал предварительно обученную модель в TF-lite и хотел бы развернуть ее на периферийном устройстве. Если мы получили новые данные для обучения и хотели бы улучшить предварительно обученную модель, можно ли это сделать на пограничном устройстве? Ex. Есть ли способ обучить модель и снова сохранить ее в TF-lite (FlatBuffer) на периферийном устройстве?

Спасибо за любые данные!

1 Ответ

1 голос
/ 25 июня 2020

Обучение на устройстве еще не полностью поддерживается в TF Lite, но вы можете обратиться к этому сообщению в блоге, чтобы узнать, как это можно сделать. https://blog.tensorflow.org/2019/12/example-on-device-model-personalization.html

Основная идея c:

  • Разделите вашу модель на базовый подграф (например, экстрактор признаков в модели классификации изображений) и обучаемая голова.
  • Преобразуйте базовый подграф в TF Lite как обычно. Преобразуйте обучаемую голову в TF Lite с помощью экспериментального инструмента tflite-transfer-convert.
  • Переобучите обучаемую голову на устройстве, как вы будете
...