分类：save_weights

2020
09-29

浅谈keras保存模型中的save()和save_weights()区别

今天做了一个关于keras保存模型的实验，希望有助于大家了解keras保存模型的区别。我们知道keras的模型一般保存为后缀名为h5的文件，比如final_model.h5。同样是h5文件用save()和save_weight()保存效果是不一样的。我们用宇宙最通用的数据集MNIST来做这个实验，首先设计一个两层全连接网络：inputs=Input(shape=(784,))x=Dense(64,activation='relu')(inputs)x=Dense(64,activation='relu')(x)y=Dense(10,activation='s... 继续阅读 >