keras中model.save和model.save_weights区别

1、保存模型model.save('**.h5')

将整个模型保存下来,以后直接载入模型与训练数据即可开始训练.

不用再定义网络和编译模型.(这种方法已经保存了模型的结构和权重,以及损失函数和优化器)

载入模型代码如下:

from keras.models import load_model

model.load_model('**.h5')

 

2、保存模型权重model.save_weights(’**.h5’)

只保存模型的权重,可以看作是保存模型的一部分.

载入模型权重代码如下:

model.load_weights('**.h5')

 

 

你可能感兴趣的:(keras)