class_docstr: The masks of a module, including the masks for weights, inputs, output
class_docstr_tok: ["The","masks","of","a","module",",","including","the","masks","for","weights",",","inputs",",","output"]
docstr: Parameters
  ----------
  name : str
      The name of the weight
  mask : CoarseMask
      The mask for this weight
docstr_tok: ["Parameters","--","--","--","--","--","name",":","str","The","name","of","the","weight","mask",":","CoarseMask","The","mask","for","this","weight"]
code: def set_param_masks(self, name, mask):
      """
      Parameters
      ----------
      name : str
          The name of the weight
      mask : CoarseMask
          The mask for this weight
      """
      self.param_masks[name] = mask
  
imports: import logging
  import torch
label_desc: Use image node_id 2 for calling the ModuleMasks obj's underlying member method code with example usage: obj.set_param_masks(name, mask) without return types