update to 0.0.15

README.md

@@ -13,7 +13,7 @@ pip3 install torch==1.8.1+cu111  torchaudio==0.8.1 -f https://download.pytorch.o
 # Installation 
 Install the latest version of this repo:
 ```shell
-pip install -e 'git+https://github.com/kkoutini/[email protected].14#egg=hear21passt' 
+pip install -e 'git+https://github.com/kkoutini/[email protected].15#egg=hear21passt' 
 ```
 
 The models follow the [common API](https://neuralaudio.ai/hear2021-holistic-evaluation-of-audio-representations.html#common-api) of HEAR 21 

hear21passt/__init__.py

@@ -1,5 +1,5 @@
 
-__version__ = "0.0.14"
+__version__ = "0.0.15"
 
 
 def embeding_size(hop=50, embeding_size=1000):

hear21passt/hop100base2lvlmel.py

@@ -34,7 +34,7 @@ def get_timestamp_embeddings(audio, model):
     """
     model.eval()
     with torch.no_grad():
-        embedmel, tmel = model.get_timestamp_mels(audio, window_size=1920)
+        embedmel, tmel = model.get_timestamp_mels(audio, window_size=6*100)
         embed1, t1 = model.get_timestamp_embeddings(audio)
         embed2, t2 = model.get_timestamp_embeddings(audio, window_size=model.timestamp_window * 5)  # larger window
         embed = torch.cat((embed1, embed2, embedmel), dim=-1)

hear21passt/hop160base2lvlmel.py

@@ -34,7 +34,7 @@ def get_timestamp_embeddings(audio, model):
     """
     model.eval()
     with torch.no_grad():
-        embedmel, tmel = model.get_timestamp_mels(audio, window_size=1920)
+        embedmel, tmel = model.get_timestamp_mels(audio, window_size=6*160) # 6 windows
         embed1, t1 = model.get_timestamp_embeddings(audio)
         embed2, t2 = model.get_timestamp_embeddings(audio, window_size=model.timestamp_window * 5)  # larger window
         embed = torch.cat((embed1, embed2, embedmel), dim=-1)