tf.keras.layers.TFSMLayer

Загрузка сохранённой модели Keras/слоя через SavedModel/ExportArchive.

Наследуется от: Layer, Operation

tf.keras.layers.TFSMLayer(
    filepath,
    call_endpoint='serve',
    call_training_endpoint=None,
    trainable=True,
    name=None,
    dtype=None
)

Пример:

model.export("path/to/artifact")
reloaded_layer = TFSMLayer("path/to/artifact")
outputs = reloaded_layer(inputs)

Загруженный объект может использоваться как обычный слой Keras и поддерживает обучение/дообучение его обучаемых весов. Обратите внимание, что загруженный объект не сохраняет внутреннюю структуру или пользовательские методы исходного объекта — это новый слой, созданный вокруг сохранённой функции.

Ограничения:

• Поддерживаются только вызовы конечных точек с одним аргументом тензора inputs (который может необязательно быть словарем/кортежем/списком тензоров). Для конечных точек с несколькими отдельными аргументами тензора ввода рассмотрите возможность наследования от TFSMLayer и реализации метода call() с пользовательской сигнатурой.
• Если вам необходимо разное поведение во время обучения и в режиме инференса (например, если вам нужен аргумент training=True в __call__() для загруженного объекта), убедитесь, что функция вызова во время обучения сохранена как автономная конечная точка в артефакте и предоставьте её имя в TFSMLayer через аргумент call_training_endpoint.

Методы

from_config

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

@classmethod
from_config(
    config
)

Создаёт слой по его конфигурации.

Этот метод — обратный к get_config, позволяющий создать тот же слой из словаря конфигурации. Он не обрабатывает соединение слоёв (обрабатывается Network), ни веса (обрабатываются set_weights).

symbolic_call

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

symbolic_call(
    *args, **kwargs
)