tensorflow/models · error · ValueError

ViT backbone is only partially loaded.

Error message

ViT backbone is only partially loaded.

What it means

Error "ViT backbone is only partially loaded." thrown in tensorflow/models.

Source

Thrown at official/projects/videoglue/tools/checkpoint_loader.py:193

        elif 'kernel' in krs_name and 'attention_output' in krs_name:
          cin, cout = ckpt_weight.shape
          num_heads = krs_w.shape[0]
          ckpt_weight = tf.reshape(ckpt_weight,
                                   [num_heads, cin // num_heads, cout])
          krs_w.assign(ckpt_weight)
          loaded.append(ckpt_name)
        else:
          skipped.append(krs_name)
      else:
        skipped.append(krs_name)

    leftover = set(ckpt_names) - set(loaded)
    logging.info('skipped: %s', skipped)
    logging.info('leftover: %s', leftover)

    if any([('encoder' in v or 'conv3d' in v or 'pos_embedding' in v)
            for v in skipped]):
      raise ValueError('ViT backbone is only partially loaded.')
    logging.info('Finished loading pretrained checkpoint from %s',
                 ckpt_dir_or_file)

  def _load_checkpoint(self, model: tf_keras.Model, ckpt_dir_or_file: str):
    """Loads checkpoint."""
    self._customized_vmae_initialize(
        model=model, ckpt_dir_or_file=ckpt_dir_or_file)


def get_checkpoint_loader(
    model: tf_keras.Model, init_checkpoint: str, init_checkpoint_type: str):
  """Gets the corresponding checkpoint loader."""

  if init_checkpoint_type == 'customized_vmae':
    return CheckpointLoaderVMAE(
        model=model,
        init_checkpoint=init_checkpoint,
        init_checkpoint_type=init_checkpoint_type)

View on GitHub (pinned to e006f5f0d5)

When it happens

Trigger: Thrown at official/projects/videoglue/tools/checkpoint_loader.py:193 when the library encounters an invalid state.

Common situations: See trigger scenarios.

AI-assisted analysis of tensorflow/models@e006f5f0d5 (2026-08-24). Data as JSON: /api/errors/cbe46ee8bcce77d6. Report an issue: GitHub.