質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

89.10%

kerasで保存モデルロード時にoptimizerの状態がロードされない

受付中

回答 0

投稿

  • 評価
  • クリップ 0
  • VIEW 90

liku

score 0

前提・実現したいこと

kerasでCycleGANを試しています。
学習時間が長いので途中でモデルを保存、そして保存したモデルをロードして学習を再開したいのですが上手くいきません。

発生している問題・エラーメッセージ

問題
  • 「モデルを保存」・「保存したモデルをロード」の動作は一応しているのですが、保存したモデルを再学習すると生成される画像がノイズのようになり、epochを繰り返しても改善しない。
  • 例えば50epoch学習時点で保存→ロードして再学習させると画像の質が下がりノイズのようになる。しかし連続して100とか200epoch学習させるとちゃんと画像の質は上がり続けている(少なくともノイズのようには全くならない)
エラーメッセージ
  • モデルロード時に以下のエラーメッセージが出ており、optimizerの状態が保存されていないようである
UserWarning: Error in loading the saved optimizer state. As a result, your model is starting with a freshly initialized optimizer.
  warnings.warn('Error in loading the saved optimizer '

該当のソースコード

discriminator = keras.models.load_model('discriminator.h5', custom_objects={'InstanceNormalization':keras_contrib.layers.InstanceNormalization})

試したこと

  • モデルをロードしてから、load_weightsで重みをロードした(optimizerとは関係ないので明らかに意味ない)
  • 上記エラーが出たときの対処方を調べたが発見できなかった
  • 保存したモデルのoptimizerの状態を確認する方法を探したが発見できなかった

補足情報

環境
  • Google Colab(GPU使用)
  • tensorflow 1.14.0
  • keras 2.1.3
  • scipy 1.1.0
  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

まだ回答がついていません

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 89.10%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

同じタグがついた質問を見る