gan_test.py 文件源码

def main():

    model = DCGAN()

    data_load = data_loader()

    with tf.Session() as sess:

        sess.run( tf.initialize_all_variables() )
        writer = tf.train.SummaryWriter('./logs', sess.graph)

        for step in range(100):

            z, image = data_load.load_next_batch()

            feed_dict = { model.z: z, model.image: image}

            sess.run([model.d_opt, model.d_opt], feed_dict = feed_dict)

            sess.run([model.g_opt], feed_dict = feed_dict)


            writer.add_summary(summary, step)