Method zero_pad

funasr/models/lora/layers.py:303–314

Zero pad. Args: x: TODO.

(self, x)

            nn.init.zeros_(self.lora_B)

    def zero_pad(self, x):
        """Zero pad.
        
            Args:
                x: TODO.
            """
        result = x.new_zeros((*x.shape[:-1], self.out_features))
        result = result.view(-1, self.out_features)
        result[:, self.lora_ind] = x.reshape(
            -1, self.out_features // len(self.enable_lora) * sum(self.enable_lora)
        )
        return result.view((*x.shape[:-1], self.out_features))

    def train(self, mode: bool = True):
        """Train.