import tensorflow as tf

    config.gpu_options.allow_growth = True
    config.allow_soft_placement = True
    
    # Train
    with tf.Session(config=config) as sess:
        try:
            summary_writer = tf.summary.FileWriter(tensorboard_dir, sess.graph)
            
