code: def restore_torch_init():
      """Rollback the change made by disable_torch_init."""
      setattr(
          torch.nn.Linear, "reset_parameters", torch_linear_init_backup
      )
      setattr(
          torch.nn.LayerNorm,
          "reset_parameters",
          torch_layer_norm_init_backup,
      )
  
file_path: deepspeedexamples/inference/huggingface/zero_inference/utils.py
imports: import torch
  import time
label_desc: Use image node_id 6 for calling a global function with example usage: restore_torch_init() without return types