北念
2023-10-12 858063b55a921d31f08582e88bdd2ba665104856
funasr/modules/frontends/mask_estimator.py
@@ -13,7 +13,7 @@
class MaskEstimator(torch.nn.Module):
    def __init__(self, type, idim, layers, units, projs, dropout, nmask=1):
        super().__init__()
        subsample = np.ones(layers + 1, dtype=np.int)
        subsample = np.ones(layers + 1, dtype=np.int32)
        typ = type.lstrip("vgg").rstrip("p")
        if type[-1] == "p":