TensorFlow-GPU 2.0で発生したいくつかの問題と解決策の記録
689 ワード
質問1:Tensorflow 2.0にh 5モデルファイルを保存した後、このh 5モデルファイルを再度読み込むことはできません.
次のコードを使用してモデルを保存します.
ただし、次のコードを使用してモデルをロードする場合:
モデルがロード中であり、メモリ使用率がデッドマシンになるまで無限に上昇しているという問題が発生しました.
問題1解決策:
2つの解決策があります
(1)tensorflowを1.14バージョンに降格し,上記の問題を解決する.
(2)モデルファイルをロードするとき、コードを次のように変更します.
load_ModelのcompileパラメータのデフォルトはTrueです.元のモデルのコンパイルプロセスが険しいため、モデルファイルとして保存してモデルをロードするときにコンパイルに問題が発生したのかもしれません.
次のコードを使用してモデルを保存します.
import tensorflow as tf
#
model.save("my_model.h5") # h5
ただし、次のコードを使用してモデルをロードする場合:
model=tf.keras.models.load_model("my_model.h5") #
モデルがロード中であり、メモリ使用率がデッドマシンになるまで無限に上昇しているという問題が発生しました.
問題1解決策:
2つの解決策があります
(1)tensorflowを1.14バージョンに降格し,上記の問題を解決する.
(2)モデルファイルをロードするとき、コードを次のように変更します.
model=tf.keras.models.load_model("my_model.h5",compile=False)
load_ModelのcompileパラメータのデフォルトはTrueです.元のモデルのコンパイルプロセスが険しいため、モデルファイルとして保存してモデルをロードするときにコンパイルに問題が発生したのかもしれません.