Function ResNet18

ML/src/python/neuralforge/models/resnet.py:4–6

(num_classes=1000, in_channels=3)

from ..nn.convolution import ResNetBlock

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

def ResNet34(num_classes=1000, in_channels=3):
    from ..nn.convolution import ResNet