import tensorflow as tf

                # scaffold_fn=tf.train.Scaffold(init_fn=train_scafflod)


        tf.logging.info("**** Trainable Variables ****")
        for var in tvars:
            init_string = ""
            if var.name in initialized_variable_names:
                init_string = ", *INIT_FROM_CKPT*"
            tf.logging.info("  name = %s, shape = %s%s", var.name, var.shape,
                            init_string)

        output_spec = None
        if mode == tf.estimator.ModeKeys.TRAIN:

            train_op = optimization_multigpus.create_optimizer(
                total_loss, learning_rate, num_train_steps, num_warmup_steps, use_tpu)
