From db15e0f46553fcde18730744ab5dfd8d6455b562 Mon Sep 17 00:00:00 2001 From: amanturamatov Date: Mon, 3 Feb 2025 17:15:01 +0600 Subject: [PATCH] Minor changes --- .../extractor/{ecapa_tdnn_taoruijie.yaml => ecapa_tdnn.yaml} | 0 oml/registry/models.py | 2 -- 2 files changed, 2 deletions(-) rename oml/configs/extractor/{ecapa_tdnn_taoruijie.yaml => ecapa_tdnn.yaml} (100%) 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 = {