tf.quantization.experimental.TfRecordRepresentativeDatasetSaver

Сохранитель представительных наборов данных в формате TFRecord.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.quantization.experimental.TfRecordRepresentativeDatasetSaver

tf.quantization.experimental.TfRecordRepresentativeDatasetSaver(
    path_map: Mapping[str, os.PathLike[str]]
)

Сохраняет представительные наборы данных для калибровки квантования в формате TFRecord. Образцы сериализуются как RepresentativeDataSample.

Метод save возвращает отображение ключа подписи к RepresentativeDatasetFile, которое может быть использовано для QuantizationOptions.

Пример использования:

# Creating the representative dataset.
representative_dataset = [{"input": tf.random.uniform(shape=(3, 3))}
                      for _ in range(256)]

# Saving to a TFRecord file.
dataset_file_map = (
  tf.quantization.experimental.TfRecordRepresentativeDatasetSaver(
        path_map={'serving_default': '/tmp/representative_dataset_path'}
    ).save({'serving_default': representative_dataset})
)

# Using in QuantizationOptions.
quantization_options = tf.quantization.experimental.QuantizationOptions(
    signature_keys=['serving_default'],
    representative_datasets=dataset_file_map,
)
tf.quantization.experimental.quantize_saved_model(
    '/tmp/input_model',
    '/tmp/output_model',
    quantization_options=quantization_options,
)

Аргументы
`path_map`	Отображение ключа определения подписи на путь. Каждый путь — это файл TFRecord, в который сохраняется `RepresentativeDataset`. Ключи определения подписи должны быть подмножеством ключей `SignatureDef` аргумента `representative_dataset` вызова `save()`.

Методы

`save`

Просмотреть исходный код

save(
    representative_dataset: RepresentativeDatasetMapping
) -> Mapping[str, _RepresentativeDatasetFile]

Сохраняет представительный набор данных.

Аргументы
`representative_dataset`	Отображение ключа определения подписи на представительный набор данных. Каждый набор данных сохраняется в отдельном файле TFRecord, путь которого соответствует ключу определения подписи `path_map`.

Исключения
`ValueError`	Если ключ определения подписи в `representative_dataset` отсутствует в `path_map`.

Возвращаемое значение
Отображение ключа подписи на экземпляр RepresentativeDatasetFile, содержащий путь к сохраненному файлу.

© 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/quantization/experimental/TfRecordRepresentativeDatasetSaver