{"record":{"id":"8c173f29de8dfb40","repo":"tensorflow/models","slug":"only-all-or-backbone-can-be-used-to-initialize","errorCode":null,"errorMessage":"Only 'all' or 'backbone' can be used to initialize the model.","messagePattern":"Only 'all' or 'backbone' can be used to initialize the model\\.","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"official/projects/centernet/tasks/centernet.py","lineNumber":168,"sourceCode":"    if not self.task_config.init_checkpoint:\n      return\n\n    ckpt_dir_or_file = self.task_config.init_checkpoint\n\n    # Restoring checkpoint.\n    if tf.io.gfile.isdir(ckpt_dir_or_file):\n      ckpt_dir_or_file = tf.train.latest_checkpoint(ckpt_dir_or_file)\n\n    if self.task_config.init_checkpoint_modules == 'all':\n      ckpt = tf.train.Checkpoint(**model.checkpoint_items)\n      status = ckpt.restore(ckpt_dir_or_file)\n      status.assert_consumed()\n    elif self.task_config.init_checkpoint_modules == 'backbone':\n      ckpt = tf.train.Checkpoint(backbone=model.backbone)\n      status = ckpt.restore(ckpt_dir_or_file)\n      status.expect_partial().assert_existing_objects_matched()\n    else:\n      raise ValueError(\n          \"Only 'all' or 'backbone' can be used to initialize the model.\")\n\n    logging.info('Finished loading pretrained checkpoint from %s',\n                 ckpt_dir_or_file)\n\n  def build_losses(self,\n                   outputs,\n                   labels,\n                   aux_losses=None):\n    \"\"\"Build losses.\"\"\"\n    input_size = self.task_config.model.input_size[0:2]\n    output_size = outputs['ct_heatmaps'][0].get_shape().as_list()[1:3]\n\n    gt_label = tf.map_fn(\n        # pylint: disable=g-long-lambda\n        fn=lambda x: target_assigner.assign_centernet_targets(\n            labels=x,\n            input_size=input_size,","sourceCodeStart":150,"sourceCodeEnd":186,"githubUrl":"https://github.com/tensorflow/models/blob/e006f5f0d534913e49c1f1dae87364039fa607e2/official/projects/centernet/tasks/centernet.py#L150-L186","documentation":"Error \"Only 'all' or 'backbone' can be used to initialize the model.\" thrown in tensorflow/models.","triggerScenarios":"Thrown at official/projects/centernet/tasks/centernet.py:168 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"}