total_time/accum_grad

游雁

2024-05-09 e299cfecaf979833d9c4d7c70e44cb92ea066afe

 funasr/frontends/utils/mask_estimator.py

@@ -23,9 +23,7 @@

        self.type = type
        self.nmask = nmask
        self.linears = torch.nn.ModuleList(
            [torch.nn.Linear(projs, idim) for _ in range(nmask)]
        )
        self.linears = torch.nn.ModuleList([torch.nn.Linear(projs, idim) for _ in range(nmask)])

    def forward(
        self, xs: ComplexTensor, ilens: torch.LongTensor