RegressionModel#
- class RegressionModel[source]#
Bases:
Module
- __init__(backbone: torch.nn.Module, out_features: int, pos_out_inds: Optional[Sequence[int]] = None, prob_out_inds: Optional[Sequence[int]] = None, **kwargs)[source]#
Methods
- __init__(backbone: torch.nn.Module, out_features: int, pos_out_inds: Optional[Sequence[int]] = None, prob_out_inds: Optional[Sequence[int]] = None, **kwargs)[source]#
- forward(x: torch.Tensor) torch.Tensor [source]#
- Parameters
x (torch.Tensor) –
- Return type