diff --git a/oml/configs/extractor/ecapa_tdnn_taoruijie.yaml b/oml/configs/extractor/ecapa_tdnn.yaml similarity index 100% rename from oml/configs/extractor/ecapa_tdnn_taoruijie.yaml rename to oml/configs/extractor/ecapa_tdnn.yaml diff --git a/oml/registry/models.py b/oml/registry/models.py index 514b0da7d..68c3e2abe 100644 --- a/oml/registry/models.py +++ b/oml/registry/models.py @@ -4,7 +4,6 @@ from torch import nn from oml.interfaces.models import IExtractor, IPairwiseModel -from oml.models.audio.ecapa_tdnn.extractor import ECAPATDNNExtractor from oml.models.meta.projection import ExtractorWithMLP from oml.models.meta.siamese import ( ConcatSiamese, @@ -23,7 +22,6 @@ "vit_clip": ViTCLIPExtractor, "vit_unicom": ViTUnicomExtractor, "extractor_with_mlp": ExtractorWithMLP, - "ecapa_tdnn": ECAPATDNNExtractor, } PAIRWISE_MODELS_REGISTRY = {