import tensorflow as tf

        """Returns a map of variables to load from a foreign checkpoint.

        Args:
          from_detection_checkpoint: whether to restore from a full detection
            checkpoint (with compatible variable names) or to restore from a
            classification checkpoint for initialization prior to training.

        Returns:
          A dict mapping variable names to variables.
        """
        return {var.op.name: var for var in tf.global_variables()}


class TrainerTest(tf.test.TestCase):

    def test_configure_trainer_and_train_two_steps(self):
        train_config_text_proto = """
    optimizer {
      adam_optimizer {
        learning_rate {
