tf.lite.RepresentativeDataset
| Просмотреть исходный код на GitHub |
Представительный набор данных, используемый для оптимизации модели.
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/versions/r2.9/api_docs/python/tf/lite/RepresentativeDataset
