Tensorflowでディープニューラルネットワークを微調整する

Question

Tensorflowで事前トレーニングされたディープニューラルネットワークを部分的に微調整したい（すべてのレイヤーをロードするが上位レイヤーのみを更新する）。

Tensorflowには、変更する必要のある変数と変更しない変数を選択できる方法はありますか？

ありがとうございます！

Answer 1

モデルをトレーニングするためのオプティマイザ（例：tf.train.AdagradOptimizer）を作成するときは、Optimizer.minimize()メソッドに明示的なvar_list=[...]引数を渡すことができます。 （このリストを指定しないと、すべての変数がデフォルトでtf.trainable_variables()に含まれます）

たとえば、モデルによっては、変数の名前を使用してリストを定義できる場合があります最適化する変数の数：