def METHOD_NAME(self):
    mod = LeNet5(10)
    batch_size = 1024
    input = torch.randn(batch_size, 1, 32, 32)
    flops, macs, params = get_model_profile(
        mod,
        tuple(input.shape),
        print_profile=True,
        detailed=True,
        module_depth=-1,
        top_modules=3,
        warm_up=1,
        as_string=False,
        ignore_modules=None,
    )
    print(flops, macs, params)
    assert within_range(flops, 866076672, TOLERANCE)
    assert within_range(macs, 426516480, TOLERANCE)
    assert params == 61706