code: def set_module_requires_grad_(module, requires_grad):
      for param in module.parameters():
          param.requires_grad = requires_grad
  
imports: import torch
  from torch import nn
  import torch.nn.functional
  from einops import rearrange, repeat
  from einops.layers.torch import Rearrange
label_desc: Use image node_id 3 for calling a global function with example usage: set_module_requires_grad_(module, requires_grad) without return types