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