Method eval

funasr/models/lora/layers.py:200–215

Eval.

(self)

            self.merged = False

    def eval(self):
        """Eval."""
        def T(w):
            """T.
            
                Args:
                    w: TODO.
                """
            return w.T if self.fan_in_fan_out else w

        nn.Linear.eval(self)
        if self.merge_weights and not self.merged:
            # Merge the weights and mark it
            if self.r > 0:
                self.weight.data += T(self.lora_B @ self.lora_A) * self.scaling
            self.merged = True

    def forward(self, x: torch.Tensor):
        """Forward pass for training.