class_docstr: A module that wraps another layer in which some weights will be replaced by 0 during training.
class_docstr_tok: ["A","module","that","wraps","another","layer","in","which","some","weights","will","be","replaced","by","0","during","training","."]
code: def reset(self):
      for layer in self.layer_names:
          raw_w = getattr(self, f"{layer}_raw")
          setattr(self.module, layer, raw_w.clone())
      if hasattr(self.module, "reset"):
          self.module.reset()
  
imports: from __future__ import annotations
  from ...data.all import *
  from ..core import *
label_desc: Use image node_id 4 for calling the WeightDropout obj's underlying member method code with example usage: obj.reset() without return types