Method __init__

src/sharp/models/encoders/monodepth_encoder.py:65–72

Initialize projection module.

(self, dims_in: list[int], dims_out: list[int])

    """Apply projection of features."""

    def __init__(self, dims_in: list[int], dims_out: list[int]) -> None:
        """Initialize projection module."""
        super().__init__()
        if len(dims_in) != len(dims_out):
            raise ValueError("Length of dims_in must be same as length of dims_out.")
        self.convs = nn.ModuleList(
            [nn.Conv2d(dim_in, dim_out, 1) for dim_in, dim_out in zip(dims_in, dims_out)]
        )

    def forward(self, encodings: list[torch.Tensor]) -> list[torch.Tensor]:
        """Apply projection module."""