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.3/api_docs/python/tf/lite/RepresentativeDataset