TensorFlow-Gpu 2.0版本中遇到的一些问题及解决方案的记录

问题1：Tensorflow2.0中保存h5模型文件后，不能再次读取该h5模型文件。

使用以下代码保存模型：

import tensorflow as tf

#模型建立代码省略
model.save("my_model.h5")  #保存h5模型文件

但当使用以下代码加载模型时：

model=tf.keras.models.load_model("my_model.h5") #加载模型

出现模型一直处于加载过程，并且内存使用率无限上涨直到死机的问题。

问题1解决办法：

有两个解决办法，

(1)，将tensorflow降级为1.14版本，上述问题解决。

(2)，在加载模型文件时,代码改为：

model=tf.keras.models.load_model("my_model.h5",compile=False)

load_model时compile参数默认为True,可能因为我的原模型编译过程比较崎岖，保存为模型文件再加载模型时编译就出现了问题吧。