tensorflow/models · error · ValueError

Stem type {} not supported.

Error message

Stem type {} not supported.

What it means

Error "Stem type {} not supported." thrown in tensorflow/models.

Source

Thrown at official/vision/modeling/backbones/resnet_unet.py:367

      x = layers.Conv2D(
          filters=int(64 * stem_depth_multiplier),
          kernel_size=3,
          strides=1,
          use_bias=False,
          padding='same',
          kernel_initializer=self._kernel_initializer,
          kernel_regularizer=self._kernel_regularizer,
          bias_regularizer=self._bias_regularizer)(
              x)
      x = self._norm(
          axis=bn_axis,
          momentum=norm_momentum,
          epsilon=norm_epsilon,
          trainable=bn_trainable)(
              x)
      x = tf_utils.get_activation(activation, use_keras_layer=True)(x)
    else:
      raise ValueError('Stem type {} not supported.'.format(stem_type))

    if replace_stem_max_pool:
      x = layers.Conv2D(
          filters=int(64 * self._depth_multiplier),
          kernel_size=3,
          strides=2,
          use_bias=False,
          padding='same',
          kernel_initializer=self._kernel_initializer,
          kernel_regularizer=self._kernel_regularizer,
          bias_regularizer=self._bias_regularizer)(
              x)
      x = self._norm(
          axis=bn_axis,
          momentum=norm_momentum,
          epsilon=norm_epsilon,
          trainable=bn_trainable)(
              x)

View on GitHub (pinned to e006f5f0d5)

Solutions

1. Use stem_type 'v0' or 'v1' in the ResNet-UNet backbone config.
2. Remove stem_type to use the default stem.

When it happens

Trigger: Thrown at official/vision/modeling/backbones/resnet_unet.py:367 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/c6a9cbe496bc8581. Report an issue: GitHub.