def METHOD_NAME():
    net = nn.Sequential(nn.BatchNorm2d(3, affine=True), nn.ReLU())
    profiler = FlopsParamsProfiler(net, torch.randn(2, 3, 12, 12))
    assert profiler.profile({}).flops == 3 * 144 * (4 + 1)
    assert profiler.profile({}).params == 6
    profiler = FlopsParamsProfiler(net, torch.randn(1, 3, 12, 12), count_normalization=False)
    assert profiler.profile({}).flops == 3 * 144 * 1
    assert profiler.profile({}).params == 0
    profiler = FlopsParamsProfiler(net, torch.randn(3, 3, 12, 12), count_activation=False)
    assert profiler.profile({}).flops == 3 * 144 * 4