テンソルフローメタからグラフを復元して別のデバイスに配置する方法

Question

メタグラフファイルから構造を復元し、展開用に配置したデバイスのみを変更するにはどうすればよいですか？

同じPythonスクリプトで、ユニットサイズ、レイヤ番号、セルタイプ（LSTMまたはGRU）などの設定可能なさまざまなパラメータを使用して、いくつかのトレーニングプロセスを開始しました。詳細。私にとって最も便利だと思うのは、モデルのデバイス配置以外のすべてを単純にインポートし、デバイスに配置して、いくつかのチェックポイントファイルで事前に訓練されたパラメータで初期化することです。

私はそれのための道を見いだせなかったので、誰かがそれに光を当てることができますか？

Answer 1

freeze_graph関数は、モデルをprotobufファイルに書き出すことができます。このファンクションにはclear_devicesというパラメータがあり、Trueに設定すると、デバイスの配置をグラフ定義から削除できます。 freeze_graphのソースコードを見てみると

手動でグラフの定義からデバイスを削除する方法を見ることができます。そのため

# Remove all the explicit device specifications for this node. This helps to 
    # make the graph more portable. 
    if clear_devices: 
    for node in input_graph_def.node: 
     node.device = "" 
    _ = tf.import_graph_def(input_graph_def, name="")