Обучение с учетом квантования в Tensorflow с использованием API высокого уровня Keras - PullRequest
5 голосов
/ 20 июня 2019

Я создал свой первый covnet, используя процесс, описанный в this colab .Теперь я хотел бы запустить модель на блестящей новой кромке ТПУ.

Но в соответствии с требованиями модели, описанными здесь , мне нужно использовать quantization-aware training (post-training quantization is not supported)., чтобы иметь возможность преобразовать модель в формат, который я могу использовать в EdgeTPU.

Как мне изменить пример колаба, чтобы выполнять эту обучающую вещь, учитывающую квантование?

1 Ответ

0 голосов
/ 28 июня 2019

хорошо, потому что API Keras не поддерживает квантование в текущей редакции, у вас есть 3 варианта:

  1. дождитесь, пока keras получит необходимую функциональность
  2. переписать вашу модель сдругой API, который имеет эту функцию
  3. , находит другой TPU, который не требует, чтобы вы квантовали ваши данные

в любом случае, хотя решение и не очень хорошее.

...