code: def set_param_grad_state(module, grad_state):
      for param in module.parameters():
          param.requires_grad = grad_state[param]
  
file_path: advertorch/advertorch/context.py
imports: from contextlib import contextmanager
label_desc: Use image node_id 6 for calling a global function with example usage: set_param_grad_state(module, grad_state) without return types