update wrapper

kkoutini · May 11, 2023 · a271296 · a271296
1 parent 17f03c8
commit a271296
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/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/passt_hear21@0.0.22#egg=hear21passt' 
+pip install -e 'git+https://github.com/kkoutini/passt_hear21@0.0.23#egg=hear21passt' 
 ```
 
 The models follow the [common API](https://neuralaudio.ai/hear2021-holistic-evaluation-of-audio-representations.html#common-api) of HEAR 21 

diff --git a/hear21passt/__init__.py b/hear21passt/__init__.py
@@ -1,5 +1,5 @@
 
-__version__ = "0.0.22"
+__version__ = "0.0.23"
 
 
 def embeding_size(hop=50, embeding_size=1000):

diff --git a/hear21passt/wrapper.py b/hear21passt/wrapper.py
@@ -5,7 +5,7 @@
 
 class PasstBasicWrapper(nn.Module):
     def __init__(self, mel: nn.Module, net: nn.Module, max_model_window=10000, timestamp_window=160, timestamp_hop=50,
-                 scene_hop=2500, scene_embedding_size=1295, timestamp_embedding_size=1295, mode="all"):
+                 scene_hop=2500, scene_embedding_size=1295, timestamp_embedding_size=1295, mode="all", **kwargs):
         """
         @param mel: spectrogram extractor
         @param net: network module