{"record":{"id":"6e8c7269cb8f07a1","repo":"tensorflow/models","slug":"teacher-model-init-checkpoint-is-not-specified-6e8c72","errorCode":null,"errorMessage":"`teacher_model_init_checkpoint` is not specified.","messagePattern":"`teacher_model_init_checkpoint` is not specified\\.","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"official/projects/mobilebert/distillation.py","lineNumber":591,"sourceCode":"    \"\"\"Checkpoints for model, stage_id, optimizer for preemption handling.\"\"\"\n    return dict(\n        stage_id=self._stage_id,\n        volatiles=self._volatiles,\n        student_pretrainer=self._student_pretrainer,\n        teacher_pretrainer=self._teacher_pretrainer,\n        encoder=self._student_pretrainer.encoder_network)\n\n  def initialize(self, model):\n    \"\"\"Loads teacher's pretrained checkpoint and copy student's embedding.\"\"\"\n    # This function will be called when no checkpoint found for the model,\n    # i.e., when the training starts (not preemption case).\n    # The weights of teacher pretrainer and student pretrainer will be\n    # initialized, rather than the passed-in `model`.\n    del model\n    logging.info('Begin to load checkpoint for teacher pretrainer model.')\n    ckpt_dir_or_file = self._task_config.teacher_model_init_checkpoint\n    if not ckpt_dir_or_file:\n      raise ValueError('`teacher_model_init_checkpoint` is not specified.')\n\n    if tf.io.gfile.isdir(ckpt_dir_or_file):\n      ckpt_dir_or_file = tf.train.latest_checkpoint(ckpt_dir_or_file)\n    # Makes sure the teacher pretrainer variables are created.\n    _ = self._teacher_pretrainer(self._teacher_pretrainer.inputs)\n    teacher_checkpoint = tf.train.Checkpoint(\n        **self._teacher_pretrainer.checkpoint_items)\n    teacher_checkpoint.read(ckpt_dir_or_file).assert_existing_objects_matched()\n\n    logging.info('Begin to copy word embedding from teacher model to student.')\n    teacher_encoder = self._teacher_pretrainer.encoder_network\n    student_encoder = self._student_pretrainer.encoder_network\n    embedding_weights = teacher_encoder.embedding_layer.get_weights()\n    student_encoder.embedding_layer.set_weights(embedding_weights)\n","sourceCodeStart":573,"sourceCodeEnd":606,"githubUrl":"https://github.com/tensorflow/models/blob/e006f5f0d534913e49c1f1dae87364039fa607e2/official/projects/mobilebert/distillation.py#L573-L606","documentation":"Error \"`teacher_model_init_checkpoint` is not specified.\" thrown in tensorflow/models.","triggerScenarios":"Thrown at official/projects/mobilebert/distillation.py:591 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"e006f5f0d534913e49c1f1dae87364039fa607e2","analyzedAt":"2026-08-24T14:09:15.576Z","schemaVersion":2},"datasetVersion":"2026-08-24T17:17:21.512Z"}