如何在Tensorflow SavedModel列出所有使用的操作?

2024-04-22 17:37:47 发布

您现在位置:Python中文网/ 问答频道 /正文

如果我使用tensorflow.saved_model保存我的模型。保存功能在SavedModel格式,我如何检索哪些Tensorflow操作是使用在这个模型之后。由于可以恢复模型,所以这些操作存储在图中,我猜是在saved_model中。pb文件。如果我加载这个protobuf(所以不是整个模型),protobuf的库部分会列出这些,但是目前还没有记录和标记这些特性。在Tensorflow 1中创建的模型。x没有这部分。

那么,从SavedModel格式的模型中检索使用过的操作列表(如MatchingFiles或WriteFile)的快速可靠方法是什么呢?

现在我可以冻结整个东西,就像tensorflowjs-converter那样。因为它们也检查所支持的操作。当LSTM在模型中时,这种方法目前不能工作,请参见这里。有没有更好的方法来做这件事,因为Ops肯定在那里?


Tags: 文件方法标记模型功能modeltensorflow格式