assignment_map
サポート
init_from_checkpoint(
ckpt_dir_or_file,
assignment_map)
あります
'checkpoint_scope_name /': 'SCOPE_NAME /' が - マッチングテンソル名でcheckpoint_scope_nameから現在SCOPE_NAME内のすべての変数をロードします。例えば
、
init_from_checkpoint('/tmp/model.ckpt',
{'old_scope_1/var1': 'new_scope_1/var1',
'old_scope_1/var2': 'new_scope_1/var2'})
は今のところ、私は2つのリスト、私はPythonで効果的に呼び出し元の関数を作るためにold_scope_1_list
とnew_scope_1_list
を使用して機能init_from_checkpoint
を呼び出すことができますどのように
old_scope_1_list=[old_scope_1/var1, old_scope_1/var2, ...,old_scope_1/var100]
new_scope_1_list=[new_scope_1/var1, new_scope_1/var2, ...,new_scope_1/var100]
がありますか?私の現在のソリューションは、それが二つのリストを使用せずに、怒鳴るなど100行を記述しているが、それはあなたがold_scope_1_listとnew_scope_1_listからペアのリストを作成するzip
functionを使用することができます
init_from_checkpoint('/tmp/model.ckpt',
{'old_scope_1/var1': 'new_scope_1/var1',
'old_scope_1/var2': 'new_scope_1/var2',
...
'old_scope_1/var100': 'new_scope_1/var100'})
の可能性のある重複した[地図二つのリストPythonで辞書に](https://stackoverflow.com/questions/209840/map- 2つのリストを辞書に入れる) – Mel
あなたは変数名のリストを持っていて、それらのそれぞれを古い/新しいスコープと連結して、必要なものを得ることができます。ものを変える場所。 –