这篇博文主要简单介绍深度学习框架keras和pytorch的模型保存方法和模型参数保存方法。
keras:
import keras
form keras.models import load_model,Sequential,Model,Input
from keras.layers import Dense,........######
model = Sequential()
model.add(Dense(units = 10,input_shape = (),name = 'dense'))
model.compile(loss = 'mse',optimizer = 'adam')
model.fit()
#########这是整个模型的保存方法
model.save('/home/路径/model.h5')
###### 导出模型
model = load_model('model.h5')# 具体路径先不写了,因为我懒
如果你只想保存模型参数的话:
#先按照上面的方法搭建一个model
model.sequential()
model.add(Dense(12),name )
model.save_weights('weights.h5')
#####想要导出weights的话
model.l