Function ResNet50

ML/src/python/neuralforge/models/resnet.py:12–15

(num_classes=1000, in_channels=3)

    return ResNet(ResNetBlock, [3, 4, 6, 3], num_classes, in_channels)

def ResNet50(num_classes=1000, in_channels=3):
    from ..nn.layers import BottleneckBlock
    from ..nn.convolution import ResNet
    return ResNet(BottleneckBlock, [3, 4, 6, 3], num_classes, in_channels)