960a107ad419d2690e352972950e76495deb1ad6,scripts/detection/ssd/eval_ssd.py,,,#,94

Before Change

    args = parse_args()

    // training contexts
    ctx = [mx.gpu(int(i)) for i in args.gpus.split(",") if i.strip()]
    ctx = ctx if ctx else [mx.cpu()]

    // network
    net_name = "_".join(("ssd", str(args.data_shape), args.network, args.dataset))
    args.save_prefix += net_name

After Change

    // eval contexts
    num_gpus = args.num_gpus
    if num_gpus > 0:
        args.batch_size *= num_gpus
    ctx = [mx.gpu(i) for i in range(num_gpus)] if num_gpus > 0 else [mx.cpu()]

    // network
    net_name = "_".join(("ssd", str(args.data_shape), args.network, args.dataset))

Instances