tf.lite.RepresentativeDataset
Набор данных, используемый для оптимизации модели.
tf.lite.RepresentativeDataset(
input_gen
)
Это функция-генератор, которая предоставляет небольшой набор данных для калибровки или оценки диапазона, т. е. (min, max) всех массивов с плавающей запятой в модели (таких как вход модели, выходные данные активации промежуточных слоёв и выход модели) для квантования. Обычно это небольшая выборка из нескольких сотен образцов, случайным образом выбранных из набора данных обучения или проверки, в произвольном порядке.
| Аргументы | |
|---|---|
input_gen | Функция-генератор, которая генерирует входные образцы для модели и имеет тот же порядок, тип и форму, что и входные данные модели. Обычно это небольшая выборка из нескольких сотен образцов, случайным образом выбранных из набора данных обучения или проверки, в произвольном порядке. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/lite/RepresentativeDataset