def recursion_change_bn(module: nn.Module) -> nn.Module:
    if isinstance(module, nn.BatchNorm2d):
        module.track_running_stats = 1
    else:
        for (i, (name, module1)) in enumerate(module._modules.items()):
            module1 = recursion_change_bn(module1)
    return module
def recursion_change_bn(module: nn.Module) -> nn.Module:
    if isinstance(module, nn.BatchNorm2d):
        module.track_running_stats = 1
    else:
        for (i, (name, module1)) in enumerate(module._modules.items()):
            module1 = recursion_change_bn(module1)
    return module