1

私は複雑なネットを持っており、そのためにモデルを推論するための非常に単純なクラスを作成し、それを凍結したグラフ ファイルにシリアル化しました。

問題は、このファイルで変数を彼の名前空間でロードする必要があることです。これは、モデルをどのように構築したかによって終了する可能性があります。私の場合、次のようになります。

# Load the input and output node with a singular namespace depending on the model
self.input_node = self.sess.graph.get_tensor_by_name("create_model/mymodelname/input_node:0")
self.output_node = self.sess.graph.get_tensor_by_name("create_model/mymodelname/out/output_node:0")

モデルに保存する前に、これら 2 つのノードにエイリアスを与えたいと思います。たとえば、最終的には共通の名前を持ち、推論用のクラスをモデルを取得するための汎用クラスとして使用できるようにします。この場合、私は次のようなことをすることになります:

# Load the input and output node with a general node namespace
self.input_node = self.sess.graph.get_tensor_by_name("input_node:0")
self.output_node = self.sess.graph.get_tensor_by_name("output_node:0")

それで、彼らにエイリアスを与えるオプションはありますか? 本当に何も見つかりませんでした。

どうもありがとうございました!

4

1 に答える 1