7e569f461b1638f3a7d0d2b825699745df876b68,tensornets/pretrained.py,,direct,#,76

Before Change

def direct(model_name, scope):
    if model_name.startswith("gen"):
        fun = load_nothing
        if "FasterRCNN" in model_name:
            if "vgg16" in scope.stem_name:
                fun = load_ref_faster_rcnn_vgg16_voc
            elif "zf" in scope.stem_name:
                fun = load_ref_faster_rcnn_zf_voc
        elif "TinyYOLOv2" in model_name:
            if "tinydarknet19" in scope.stem_name:
                fun = load_ref_tiny_yolo_v2_voc
        elif "YOLOv2" in model_name:
            if "darknet19" in scope.stem_name:
                fun = load_ref_yolo_v2_voc
    else:
        fun = __load_dict__[model_name]

    def _direct():

After Change

def direct(model_name, scope):
    if model_name.startswith("gen"):
        model_name = model_name[3:].lower()
        stem_name = scope.stem.model_name
        try:
            fun = __gen_load_dict__[model_name][stem_name]
        except KeyError:
            fun = load_nothing
            warnings.warn("Random initialization will be performed because "
                          "the pre-trained weights for " + model_name +

Instances