def METHOD_NAME(num_channels):
    if dist.get_world_size() > 1:
        return nn.SyncBatchNorm.convert_sync_batchnorm(nn.BatchNorm1D(num_channels))
    else:
        return nn.BatchNorm1D(num_channels)