main.py 文件源码

def main(_):
    with tf.Session(config=tf.ConfigProto(log_device_placement=FLAGS.log_dev_placement)) as sess:
        dcgan = DCGAN(sess, batch_size=FLAGS.batch_size,
                      #in_dim=[28,28,1], z_dim=100)
                      in_dim=[112,112,3], z_dim=100)
        dcgan.train(FLAGS)