2318052dc79966bf36675606b7d992a347418292,scripts/detection/yolo/train_yolo3.py,,,#,295

Before Change

    // use sync bn if specified
    num_sync_bn_devices = len(ctx) if args.syncbn else -1
    if num_sync_bn_devices > 1:
        net = get_model(net_name, pretrained_base=True, num_sync_bn_devices=num_sync_bn_devices)
        async_net = get_model(net_name, pretrained_base=False)  // used by cpu worker
    else:
        net = get_model(net_name, pretrained_base=True)
        async_net = net

After Change

    args.save_prefix += net_name
    // use sync bn if specified
    if args.syncbn and len(ctx) > 1:
        net = get_model(net_name, pretrained_base=True, norm_layer=gluon.contrib.nn.SyncBatchNorm,
                        norm_kwargs={"num_devices": len(ctx)})
        async_net = get_model(net_name, pretrained_base=False)  // used by cpu worker
    else:
        net = get_model(net_name, pretrained_base=True)
        async_net = net

Instances