tensorflow2.x 如何用已经保存的模型参数直接预测

Question

1、第一次已经训练好网络并保存了它的参数，第二次不训练网络直接用保存的参数进行预测(前后两次的网络结构一模一样）

2、第一次：

model.save_weights('C:/TF2.1/base_model.h5')

3、第二次：

model.load_weights('base_model.h5')

model.predict(x_test)

4、报错：You are trying to load a weight file containing 4 layers into a model with 0 layers.

Answer 1

用load_model加载模型
不要调用 fit
直接调用 predict