srez_train.py 文件源码

def test_model(train_data):
    td = train_data

    summaries = tf.summary.merge_all()
    td.sess.run(tf.initialize_all_variables())
    start_time = time.time()

    batch = 0
    done = False

    while not done:
        print("Processing batch {0}".format(batch))
        test_feature, test_label = td.sess.run([td.features, td.labels])

        # Show progress with test features
        feed_dict = {td.gene_minput: test_feature}
        gene_output = td.sess.run(td.gene_moutput, feed_dict=feed_dict)
        _save_output(td, test_feature, test_label, gene_output, batch, 'out')
        batch += 1

    print('Finished testing!')