我拥有定制的预训练模型的权重。我需要在模型的不同层中提取通过模型的不同输入的表示形式。最好的方法是什么？
我正在使用TensorFlow 2.1.0，目前使用hub.KerasLayer（）或tf.saved_model.load（）加载模型的权重>
任何帮助将不胜感激！我是TensorFlow的新手，别无选择，只能使用它，因为权重是从其他来源获得的。
我拥有定制的预训练模型的权重。我需要在模型的不同层中提取通过模型的不同输入的表示形式。最好的方法是什么...
tf.saved_model.load（）及其包装器中心。KerasLayer加载计算图和预训练的权重。我想您正在处理一个TF2样式的SavedModel，其计算打包在TensorFlow函数中。如果是这样，没有简单的方法可以从函数中提取中间结果。如果可能，您可以要求模型创建者提供更多输出，或者，如果您具有模型的Python源，则从源构建模型，并使用SavedModel的权重初始化其权重（需要一些测量）。