aky15
2023-04-14 fa25b637b0d257186a8399eb1c530a91f4252702
funasr/modules/repeat.py
@@ -49,13 +49,12 @@
        block_list: List[torch.nn.Module],
        output_size: int,
        norm_class: torch.nn.Module = torch.nn.LayerNorm,
        norm_args: Optional[Dict] = None,
    ) -> None:
        """Construct a MultiBlocks object."""
        super().__init__()
        self.blocks = torch.nn.ModuleList(block_list)
        self.norm_blocks = norm_class(output_size, **norm_args)
        self.norm_blocks = norm_class(output_size)
        self.num_blocks = len(block_list)