resnet_50¶
Overview¶
The resnet_50 function constructs a ResNet-50 model, a high-performance residual network suitable for a wide range of image classification tasks.
It uses the preset ResNetConfig(block=”bottleneck”, layers=[3, 4, 6, 3]) and accepts additional ResNetConfig keyword overrides such as in_channels, stem_type, stem_width, avg_down, channels, and block_args.
Total Parameters: 25,557,032
Function Signature¶
@register_model
def resnet_50(num_classes: int = 1000, **kwargs) -> ResNet:
Parameters¶
num_classes (int, optional): Number of output classes for the classification task. Default is 1000.
kwargs: Additional keyword arguments forwarded to ResNetConfig, excluding the preset block and layers fields.
Returns¶
ResNet: An instance of the ResNet-50 model.
Examples¶
Creating a ResNet-50 model for 1000 classes:
model = resnet_50(num_classes=1000)
print(model)
Note
ResNet-50 uses Bottleneck with a stage configuration of [3, 4, 6, 3].
The returned model is equivalent to ResNet(ResNetConfig(…)) with the preset values above.