変数とその状態をチェックポイントから読み取るにはどうすればよいですか?Tensorflow。チェックポイントのリスト変数
私は自動エンコーダで作業しています。チェックポイントには、エンコーダ、デコーダ、オプティマイザなどのネットワークの完全な状態が含まれています。エンコーディングをばかにしたいので、評価モードでネットワークのデコーダ部分のみが必要です。
より抽象的な方法で同じ質問:別のモデルで再利用するために既存のチェックポイントから特定の変数のみを読み取るにはどうすればよいですか?
私は変数に対応して名前を付けるべきですか? かのようなものを取得する方法があります:
w_init = read_from_state(state_location, var_name)
def read_from_state(state_location, var_name):
# the magic goes here
pass