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]#
Parameters

Methods

__init__(backbone_arch, out_features[, ...])

forward(x)

__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]#
Parameters
static __new__(cls, *args: Any, **kwargs: Any) → Any#
Parameters

  • args (Any) –

  • kwargs (Any) –

Return type

Any

forward(x: torch.Tensor) → torch.Tensor[source]#
Parameters

x (torch.Tensor) –

Return type

torch.Tensor