class_docstr: A hook to process sharded param before and after FWD and BWD operator executing.
class_docstr_tok: ["A","hook","to","process","sharded","param","before","and","after","FWD","and","BWD","operator","executing","."]
code: def post_bwd_exec(self, module: torch.nn.Module, input):
      pass
  
file_path: ColossalAI/colossalai/legacy/zero/gemini/ophooks/_shard_grad_ophook.py
imports: import torch
  from colossalai.legacy.registry import OPHOOKS
  from .None import BaseOpHook
label_desc: Use image node_id 5 for calling the ShardGradMemTracerHook obj's underlying member method code with example usage: obj.post_bwd_exec(module, input) without return types