Tensorflow学习七---模型的保存与加载

Tensorflow学习七—模型的保存与加载

模型的保存

模型的保存是很有必要的，在复杂的网络训练时间很长的时候，为了防止意外出现所以合理的保存模型是很有必要的。

保存模型分为三种模式
save/load weights
最轻量级，干净的保存模型的方式。只保存网络参数，其他信息不管。

#save
model.save_weights('./checkpoints/my_checkpoints')
#restore
model = create_mode()
model.load_weights('./checkpoints/my_checkpoints')
loss,acc = model.evaluate(test_images,test_labels)

save/load entire model
将所有状态都保存起来

network.save('model.h5')
printf("save")
del network