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`	Отображение ключей signature def на пути. Каждый путь — это файл TFRecord, в который сохраняется `RepresentativeDataset`. Ключи signature def должны быть подмножеством ключей `SignatureDef` аргумента `representative_dataset` вызова `save()`.

Методы

`save`

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

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

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

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

Вызывает исключения
`ValueError`	Если ключ signature def в `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