Help us understand the problem. What is going on with this article?

【keras】重たいモデルのload/save

More than 1 year has passed since last update.

kerasでモデルを学習して保存しておく際、通常であれば

model.save('test_model.h5')

などで保存するのが一般的であるが、これをそのままload_modelすると以下のようなエラーが出ることがある。100M以上とかの大きいファイルを読み込む時に良く出て来る。

OSError: Unable to open file (file signature not found)

保存したモデルを推論に使うのみであれば、以下のように学習に必要な情報を省略して保存することで、ファイルの大きさが半分程度になる。大体の場合はこれでエラーが出なくなる。

model.save('test_model.h5',  include_optimizer=False)
Gri_Gra
若手技術者です。仕事では主にデータ解析とかやっています。趣味で作るものは仕事に関係あったりなかったり。pythonや機械学習まわりのことを備忘録的に書き留めてこうと思います。 記事に修正点/改善点等あればコメントください。
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away