Function ResNet34

ML/src/python/neuralforge/models/resnet.py:8–10

(num_classes=1000, in_channels=3)

    return ResNet(ResNetBlock, [2, 2, 2, 2], num_classes, in_channels)

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

def ResNet50(num_classes=1000, in_channels=3):
    from ..nn.layers import BottleneckBlock