嘉渊
2023-05-11 6737c14fff2a23cf4cc7d2ae6d5c3bf4a5d12c98
funasr/models/encoder/ecapa_tdnn_encoder.py
@@ -528,8 +528,6 @@
    Arguments
    ---------
    device : str
        Device used, e.g., "cpu" or "cuda".
    activation : torch class
        A class for constructing the activation layers.
    channels : list of ints
@@ -555,7 +553,6 @@
    def __init__(
        self,
        input_size,
        device="cpu",
        lin_neurons=192,
        activation=torch.nn.ReLU,
        channels=[512, 512, 512, 512, 1536],