#模型的读写
#官方模型保存参数和权重(.h5模型)
model.save( filepath="model.h5", overwrite=True, include_optimizer=True )
model=keras.models.load_model( "model.h5" )
#自定义模型只保存参数(model_weight模型)
model.save_weights('model_weight')
model.load_weights('model_weight')
【keras】API笔记
最新推荐文章于 2024-05-09 20:43:47 发布