diff --git a/funasr/models/encoder/resnet34_encoder.py b/funasr/models/encoder/resnet34_encoder.py index 6f978eb34..df0f124ca 100644 --- a/funasr/models/encoder/resnet34_encoder.py +++ b/funasr/models/encoder/resnet34_encoder.py @@ -2,6 +2,7 @@ import torch from torch.nn import functional as F from typing import Tuple, Optional from funasr.models.pooling.statistic_pooling import statistic_pooling, windowed_statistic_pooling +from funasr.models.encoder.abs_encoder import AbsEncoder from collections import OrderedDict import logging import numpy as np