hnluo
2023-03-10 1eacb8ae811a7e68c296523e6a8b4e72eb024aa2
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],