RegressionModel#
- class RegressionModel[source]#
Bases:
Module
- __init__(backbone_arch: str, out_features: int, pretrained: bool = True, pos_out_inds: Optional[Sequence[int]] = None, prob_out_inds: Optional[Sequence[int]] = None, **kwargs)[source]#
Methods
- __init__(backbone_arch: str, out_features: int, pretrained: bool = True, 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