tensorflow 保存模型 加载模型

1、训练的时候 保存模型不要太大，太大restore会有问题

 

saver = tf.train.Saver()

 

if FLAGS.reloading:

        # Initialize variables
        session.run(tf.global_variables_initializer())
        # Initialize target network weights
        session.run(graph_ops["reset_target_network_params"])

        saver = tf.train.import_meta_graph(FLAGS.metadata_path)
        saver.restore(session, FLAGS.checkpoint_path)
    else:
        # Initialize variables
        session.run(tf.global_variables_initializer())
        # Initialize target network weights
        session.run(graph_ops["reset_target_network_params"])

 

 

按顺序写这两句就可以加载了