9f0b230e1f59cb399e36210483bc075760b02eaf,scripts/detection/ssd/eval.py,,,#,70

Before Change

    // network
    net_name = "_".join(("ssd", str(args.data_shape), args.network))
    net = get_model(net_name, classes=len(classes), pretrained=0)  // load pretrained base network
    net.load_params(args.pretrained.strip())

    // training

After Change

    // network
    net_name = "_".join(("ssd", str(args.data_shape), args.network, args.dataset))
    if args.pretrained.lower() in ["true", "1", "yes", "t"]:
        net = gv.model_zoo.get_model(net_name, pretrained=True)
    else:
        net = gv.model_zoo.get_model(net_name, pretrained=False)

Instances