yolov1's backbone network resnet has a bit of a problem

[ThreeStones1029]

I found several mistakes in the backbone code of yolov1. The code error is in models/detectors/yolov1/yolov1_backbone.py 232 and 235,maybe you forgot to change it,duplicate resnet34 appears.
Before modification

def build_backbone(model_name='resnet18', pretrained=False):
    if model_name == 'resnet18':
        model = resnet18(pretrained)
        feat_dim = 512
    elif model_name == 'resnet34':
        model = resnet34(pretrained)
        feat_dim = 512
    elif model_name == 'resnet50':
        model = resnet34(pretrained)
        feat_dim = 2048
    elif model_name == 'resnet101':
        model = resnet34(pretrained)
        feat_dim = 2048

After modification

def build_backbone(model_name='resnet18', pretrained=False):
    if model_name == 'resnet18':
        model = resnet18(pretrained)
        feat_dim = 512
    elif model_name == 'resnet34':
        model = resnet34(pretrained)
        feat_dim = 512
    elif model_name == 'resnet50':
        model = resnet50(pretrained)
        feat_dim = 2048
    elif model_name == 'resnet101':
        model = resnet101(pretrained)
        feat_dim = 2048