packed_wideresnet28x10

torch_uncertainty.models.packed_wideresnet28x10(in_channels, num_classes, num_estimators, alpha, gamma, conv_bias=True, dropout_rate=0.3, groups=1, style='imagenet', activation_fn=<function relu>, normalization_layer=<class 'torch.nn.modules.batchnorm.BatchNorm2d'>)

Packed-Ensembles of Wide-ResNet-28x10.

Parameters:

• in_channels (int) – Number of input channels.

• num_classes (int) – Number of classes to predict.

• num_estimators (int) – Number of estimators in the ensemble.

• alpha (int) – Expansion factor affecting the width of the estimators.

• gamma (int) – Number of groups within each estimator.

• groups (int) – Number of subgroups in the convolutions.

• conv_bias (bool) – Whether to use bias in convolutions. Defaults to True.

• dropout_rate (float, optional) – Dropout rate. Defaults to 0.3.

• style (bool, optional) – Whether to use the ImageNet structure. Defaults to True.

• activation_fn (Callable, optional) – Activation function. Defaults to torch.nn.functional.relu.

• normalization_layer (nn.Module, optional) – Normalization layer. Defaults to torch.nn.BatchNorm2d.

Returns:

A Packed-Ensembles Wide-ResNet-28x10.

Return type:

_PackedWideResNet