keras模型保存和權重保存

模型保存和讀取(包括權重):python

model.save('./model.h5')

from keras import models
model = models.load_model(./model.h5) # 讀取以前不須要從新定義網絡框架

模型權重的保存和讀取:網絡

model.save_weights('./weights.h5')

model.load_weights('./weights.h5') # 因爲這個只是保存了權重,因此在這句話以前還要從新定義網絡
相關文章
相關標籤/搜索