<gh_stars>1-10
import torchvision.models as models


def getResNet(size, pretrained=False, model_args=None):
    if size == 50:
        return models.resnet50(pretrained=pretrained)
    elif size == 34:
        return models.resnet34(pretrained=pretrained)
    elif size == 101:
        return models.resnet101(pretrained=pretrained)
    elif size == 18:
        return models.resnet18(pretrained=pretrained)

import torchvision.models as models


def getResNet(size, pretrained=False, model_args=None):
    if size == 50:
        return models.resnet50(pretrained=pretrained)
    elif size == 34:
        return models.resnet34(pretrained=pretrained)
    elif size == 101:
        return models.resnet101(pretrained=pretrained)
    elif size == 18:
        return models.resnet18(pretrained=pretrained)