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"])
按顺序写这两句就可以加载了