tensorflow 模型加载(没有checkpoint文件或者说只加载其中一个模型)

1、如果有checkpoint文件的话,加载模型很简单:

  第一步:都是加载图:

with tf.Session() as sess: 
  saver=tf.train.import_meta_graph('./insightface_iter_best_71000.ckpt.meta')   saver.restore(sess,tf.train.latest_checkpoint('./'))

 

2、没有该文件,只有一个模型的时候:

with tf.Session() as sess: 
  saver=tf.train.import_meta_graph('./insightface_iter_best_71000.ckpt.meta')
  saver.restore(sess,'./insightface_iter_best_71000.ckpt')  # 重点,将地址写到.ckpt

 

上一篇:spark(三):blockManager、broadcast、cache、checkpoint


下一篇:Checkpoint防火墙