def METHOD_NAME():
    """
    Code example for model visualization
    """
    model = models.resnet18(pretrained=True).to(torch.device('cpu'))
    model = model.eval()
    # Specify a folder in which the plots will be saved
    results_dir = './visualization'
    batch_norm_fold.fold_all_batch_norms(model, (1, 3, 224, 224))
    # Usually it is observed that if we do BatchNorm fold the layer's weight range increases.
    # This helps in finding layers which can be equalized to get better performance on hardware
    visualize_model.visualize_relative_weight_ranges_to_identify_problematic_layers(model, results_dir)