我如何离线加载hub.KerasLayer?

问题描述 投票:0回答:1

我已经通过tensorflow hub建立了一个模型并保存。但是,当我加载它时,我必须添加custom_objects={'KerasLayer':hub.KerasLayer}。它将连接网络。

model = tf.keras.models.load_model('my_model.h5',custom_objects={'KerasLayer':hub.KerasLayer})

如何下载hub.KerasLayer并离线加载?

tensorflow2.0 keras-layer h5py tensorflow-hub
1个回答
0
投票

其中有些取决于您是否具有正确保存的模型。在这种情况下,您可以]

import tensorflow as tf
import tensorflow_hub as hub


k_layer = hub.KerasLayer("some/file/path")

有关更多详细信息,请参见https://www.tensorflow.org/hub/tf2_saved_model

© www.soinside.com 2019 - 2024. All rights reserved.