{"record":{"id":"cbe46ee8bcce77d6","repo":"tensorflow/models","slug":"vit-backbone-is-only-partially-loaded","errorCode":null,"errorMessage":"ViT backbone is only partially loaded.","messagePattern":"ViT backbone is only partially loaded\\.","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"official/projects/videoglue/tools/checkpoint_loader.py","lineNumber":193,"sourceCode":"        elif 'kernel' in krs_name and 'attention_output' in krs_name:\n          cin, cout = ckpt_weight.shape\n          num_heads = krs_w.shape[0]\n          ckpt_weight = tf.reshape(ckpt_weight,\n                                   [num_heads, cin // num_heads, cout])\n          krs_w.assign(ckpt_weight)\n          loaded.append(ckpt_name)\n        else:\n          skipped.append(krs_name)\n      else:\n        skipped.append(krs_name)\n\n    leftover = set(ckpt_names) - set(loaded)\n    logging.info('skipped: %s', skipped)\n    logging.info('leftover: %s', leftover)\n\n    if any([('encoder' in v or 'conv3d' in v or 'pos_embedding' in v)\n            for v in skipped]):\n      raise ValueError('ViT backbone is only partially loaded.')\n    logging.info('Finished loading pretrained checkpoint from %s',\n                 ckpt_dir_or_file)\n\n  def _load_checkpoint(self, model: tf_keras.Model, ckpt_dir_or_file: str):\n    \"\"\"Loads checkpoint.\"\"\"\n    self._customized_vmae_initialize(\n        model=model, ckpt_dir_or_file=ckpt_dir_or_file)\n\n\ndef get_checkpoint_loader(\n    model: tf_keras.Model, init_checkpoint: str, init_checkpoint_type: str):\n  \"\"\"Gets the corresponding checkpoint loader.\"\"\"\n\n  if init_checkpoint_type == 'customized_vmae':\n    return CheckpointLoaderVMAE(\n        model=model,\n        init_checkpoint=init_checkpoint,\n        init_checkpoint_type=init_checkpoint_type)","sourceCodeStart":175,"sourceCodeEnd":211,"githubUrl":"https://github.com/tensorflow/models/blob/e006f5f0d534913e49c1f1dae87364039fa607e2/official/projects/videoglue/tools/checkpoint_loader.py#L175-L211","documentation":"Error \"ViT backbone is only partially loaded.\" thrown in tensorflow/models.","triggerScenarios":"Thrown at official/projects/videoglue/tools/checkpoint_loader.py:193 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"}