import tensorflow as tf


        with tf.device(FLAGS.device):
            lr = tf.placeholder(tf.float32, shape=[], name="learning_rate")
            mom = tf.placeholder(tf.float32, shape=[], name="momentum")
            with tf.variable_scope("CNN") as scope:
                # Build training graph
                loss, train_op, global_step, ul_u_updated = build_training_graph(
                                                                images, labels, ul_images, ul_u, lr, mom)
