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.