class_docstr: BaseOffloadModule: A model wrapper for parameter offloading.
  
  Args:
      model (nn.Module): model to apply offloading.
      region_manager (RegionManager): a ``RegionManager`` instance.
      is_sync (bool): synchronous mode or not.
class_docstr_tok: ["BaseOffloadModule",":","A","model","wrapper","for","parameter","offloading",".","Args",":","model","(","nn.Module",")",":","model","to","apply","offloading",".","region_manager","(","RegionManager",")",":","a","``","RegionManager","``","instance",".","is_sync","(","bool",")",":","synchronous","mode","or","not","."]
code: def remove_grad_hook(self):
      for hook in self.grad_hook_list:
          hook.remove()
  
imports: from functools import partial
  from typing import Optional, Set
  import torch
  import torch.nn
  from colossalai.utils import _cast_float
  from colossalai.zero.legacy.gemini.tensor_utils import free_storage
  from .region_manager import RegionManager
  from .util import GlobalRuntimeInfo
label_desc: Use image node_id 3 for calling the BaseOffloadModule obj's underlying member method code with example usage: obj.remove_grad_hook() without return types