Class GaussianDecoderParams

src/sharp/models/params.py:132–160

Parameters for backbone with default values.

@dataclasses.dataclass
class GaussianDecoderParams:
    """Parameters for backbone with default values."""

    dim_in: int = 5
    dim_out: int = 32
    # Which normalization to use in backbone.
    norm_type: NormLayerName = "group_norm"
    # How many groups to use for group normalization.
    norm_num_groups: int = 8
    # Stride of backbone.
    stride: int = 2

    patch_encoder_preset: ViTPreset = "dinov2l16_384"
    image_encoder_preset: ViTPreset = "dinov2l16_384"

    # Dimensionality of feature maps for DPT decoder.
    dims_decoder: DimsDecoder = (128, 128, 128, 128, 128)

    # Whether to use depth as input.
    use_depth_input: bool = True

    # Whether to enable gradient checkpointing for the backbone
    grad_checkpointing: bool = False

    # What mode to use for upsampling in decoder.
    upsampling_mode: UpsamplingMode = "transposed_conv"

    # The type of image encoder.
    image_encoder_type: DPTImageEncoderType = "skip_conv_kernel2"


@dataclasses.dataclass