Poolings

Classification

class torchok.models.poolings.classification.linear.PoolingLinear(in_channels, out_channels, pooling_type: str = 'avg', output_size: int = 1, bias=True)

Bases: Pooling

__init__(in_channels, out_channels, pooling_type: str = 'avg', output_size: int = 1, bias=True)

forward(x: Tensor) → Tensor

init_weights()

training: bool

class torchok.models.poolings.classification.pooling.Pooling(in_channels: int, pooling_type: str = 'avg', output_size: int = 1)

Bases: SelectAdaptivePool2d, BaseModel

__init__(in_channels: int, pooling_type: str = 'avg', output_size: int = 1)

training: bool