code: def ignore_bn(ws: Dict):
      ignored_ws = {}
      for k in ws:
          if len(ws[k].size()) < 2:
              ignored_ws[k] = torch.zeros(
                  size=ws[k].size(), device=ws[k].device
              )
          else:
              ignored_ws[k] = ws[k]
      return ignored_ws
  
file_path: ml-cvnets/loss_landscape/landscape_utils.py
imports: import copy
  from typing import Dict, Optional
  import matplotlib.pyplot
  import numpy
  import torch
  from matplotlib import animation, cm
  from mpl_toolkits.mplot3d import Axes3D
  from utils.common_utils import unwrap_model_fn
label_desc: Use image node_id 3 for calling a global function with example usage: ignore_bn(ws) and returns: ignored_ws