2016-09-13 24 views
4

私はいくつかの入力データで勾配降下(GD)を使ってうまく練習したテンソルフローモデル(畳み込みニューラルネットワーク)を持っています。Tensorflow:勾配降下で入力を最適化

ここで、2番目のステップでは、入力画像を初期化し、GDを使用して固定ネットワークパラメータでこの入力画像を最適化したいと考えています。損失関数は異なるものですが、これは詳細です。ネットワークパラメータ の最適化を停止 - - 最初は、おそらくこの Holding variables constant during optimizer

で行うことができ、入力画像

上で最適化するために、

だから、私の主な質問は に勾配降下アルゴリズムに伝える方法です皆さんは2番目のポイントについて考えていますか?

私はグラジエント降下アルゴリズムを自分自身でTF勾配関数を使用してコード化することができますが、私の腸の感覚は、より複雑なGDバリアント(アダムなど) 。

ありがとうございます! クリスチャン

+0

答えを見つけましたか? – neouyghur

+0

いいえ、私はしませんでした。私は実際に自分でSGDアルゴリズムをコーディングしました。 – Flonks

答えて

1
  1. trainable=Truetf.Variableとして画像を表す trainable=FalseとTF変数を用いてNNグラフを再作成し、訓練されたNNのグラフから重みをコピー
  2. 開始画像(初期推定値)と、この変数を初期化を使用してtf.assign
  3. 損失関数を計算する
  4. 任意のTF最適化アルゴリズムに損失をプラグ
+0

ありがとうございます。 SGDの自分のバージョンを使用して解決しました。その後、PyTorchに切り替えました。 – Flonks

関連する問題