|
@@ -2,6 +2,7 @@ import torch
|
|
|
from torch.nn import functional as F
|
|
from torch.nn import functional as F
|
|
|
from typing import Tuple, Optional
|
|
from typing import Tuple, Optional
|
|
|
from funasr.models.pooling.statistic_pooling import statistic_pooling, windowed_statistic_pooling
|
|
from funasr.models.pooling.statistic_pooling import statistic_pooling, windowed_statistic_pooling
|
|
|
|
|
+from funasr.models.encoder.abs_encoder import AbsEncoder
|
|
|
from collections import OrderedDict
|
|
from collections import OrderedDict
|
|
|
import logging
|
|
import logging
|
|
|
import numpy as np
|
|
import numpy as np
|