util.py 文件源码

def initialize_session(sess, task_params):
    if task_params['verbose']:
        print("Initalizing tensorflow session ...")

    saver = tf.train.Saver()
    if task_params['restore_from_checkpoint']:
        saver.restore(
            sess=sess,
            save_path=task_params['save_path'])
        if task_params['verbose']:
            print("Restoring variables from '{}'".format(task_params['save_path']))
    else:
        sess.run(tf.initialize_all_variables())
        sess.run(tf.initialize_local_variables())

    coord = tf.train.Coordinator()
    threads = tf.train.start_queue_runners(sess=sess, coord=coord)

    logdir=task_params['summaries_dir'] + '/train_' + time.strftime("%Y%m%d_%H-%M-%S")
    train_writer = tf.train.SummaryWriter(logdir=logdir, graph=sess.graph)

    summaries = tf.merge_all_summaries()


    return coord, threads, saver, train_writer, summaries