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