断点续训

作用:存取模型

方法:

  读取模型: 

1 checkpoint_save_path = './checkpoint/Baseline.ckpt'
2 if os.path.exists(checkpoint_save_path+'.index'):
3     print('load exists model')
4     model.load_weights(checkpoint_save_path)

  保存模型:

  

posted @ 2020-07-24 14:20  要想瘦先吃肉  阅读(144)  评论(0编辑  收藏  举报