code: def _setweights(self):
          "Apply dropout to the raw weights."
          for layer in self.layer_names:
              raw_w = getattr(self, f'{layer}_raw')
              self.module._parameters[layer] = F.dropout(raw_w, p=self.weight_p, training=self.trainingf _setweights(self):
          "Apply dropout to the raw weights."
          for layer in self.layer_names:
              raw_w = getattr(self, f'{layer}_raw')
              self.module._parameters[layer] = F.dropout(raw_w, p=self.weight_p, training=self.training)
docs: Apply dropout to the raw weights.