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