add SC w2v-bert 2.0

nguyenvulong · Dec 24, 2023 · 16d3c23 · 16d3c23
1 parent a1870cf
commit 16d3c23
Show file tree

Hide file tree

Showing 4 changed files with 663 additions and 30 deletions.
diff --git a/models/__init__.py b/models/__init__.py
@@ -1,3 +1,3 @@
-from .lcnn import *
-from .senet import *
-from .occm import *
+# from .lcnn import *
+# from .senet import *
+# from .occm import *
diff --git a/models/sslassist.py b/models/sslassist.py
@@ -21,7 +21,7 @@ class SSLModel(nn.Module):
     def __init__(self,device):
         super(SSLModel, self).__init__()
 
-        cp_path = '/datac/longnv/SSL_Anti-spoofing/pretrained/xlsr2_300m.pt'   # Change the pre-trained XLSR model path. 
+        cp_path = '/home/longnv/BTS-Encoder-ASVspoof/demo2023/pretrained/xlsr2_300m.pt'   # Change the pre-trained XLSR model path. 
         model, cfg, task = fairseq.checkpoint_utils.load_model_ensemble_and_task([cp_path])
         self.model = model[0]
         self.device=device