import torch
import functools

if torch.__version__.startswith('0'):
    from .sync_bn.inplace_abn.bn import InPlaceABNSync
    BatchNorm2d = functools.partial(InPlaceABNSync, activation='none')
    BatchNorm2d_class = InPlaceABNSync
    relu_inplace = False
else:
    # BatchNorm2d_class = BatchNorm2d = torch.nn.SyncBatchNorm
    BatchNorm2d_class = BatchNorm2d = torch.nn.BatchNorm2d
    relu_inplace = True
[((34, 67), 'torch.__version__.startswith', 'torch.__version__.startswith', (['"""0"""'], {}), "('0')\n", (62, 67), False, 'import torch\n'), ((142, 194), 'functools.partial', 'functools.partial', (['InPlaceABNSync'], {'activation': '"""none"""'}), "(InPlaceABNSync, activation='none')\n", (159, 194), False, 'import functools\n')]