def plot_gradflow_v1(named_parameters):
    ...

def plot_gradflow_v2(named_parameters):
    ...

    Usage: Plug this function in Trainer class after loss.backwards() as "plot_grad_flow(self.model.named_parameters())" to visualize the gradient flow'''
                                             ...


