4a53e30fafdd8c65c42d0cf978312e6333182db4,modules/functions.py,InPlaceABNSync,forward,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,167

Before Change

        n = _count_samples(x) * (ctx.master_queue.maxsize + 1)

        if ctx.training:
            mean = x.new().resize_(1, running_mean.size(0))
            var = x.new().resize_(1, running_var.size(0))
            _check_contiguous(x, mean, var)
            _check(_ext.bn_mean_var_cuda, x, mean, var)

After Change

        count = _count_samples(x) * (ctx.master_queue.maxsize + 1)
        x = x.contiguous()
        weight = weight.contiguous() if ctx.affine else x.new_empty(0)
        bias = bias.contiguous() if ctx.affine else x.new_empty(0)

        if ctx.training:
            mean, var = backend.mean_var(x)

Instances