code: def restore_torch_init():
      """Rollback the change made by disable_torch_init."""
      import torch
  
      setattr(
          torch.nn.Linear, "reset_parameters", torch_linear_init_backup
      )
      setattr(
          torch.nn.LayerNorm,
          "reset_parameters",
          torch_layer_norm_init_backup,
      )
  
file_path: H2O/h2o_flexgen/flexgen/opt_config.py
imports: import argparse
  import dataclasses
  import glob
  import os
  import shutil
  import numpy
  from tqdm import tqdm
label_desc: Use image node_id 4 for calling a global function with example usage: restore_torch_init() without return types