fix bug: config help

metaphysicser · Feb 10, 2023 · 4c274f8 · 4c274f8
1 parent 095f8a2
commit 4c274f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config.py b/config.py
@@ -95,7 +95,7 @@ def get_config(parse=True, **optional_kwargs):
     parser.add_argument('--scale_weight', type=float, default=0.1)
 
     parser.add_argument('--model', type=str,
-                        default='PS-Mixer', help='one of {PS-Mixer, }')
+                        default='PS_Mixer', help='one of {PS_Mixer, }')
 
     parser.add_argument('--test_duration', type=int, default=1)
 

diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ audtorch==0.6.4
 colorama==0.4.4
 einops==0.4.1
 gensim==4.1.2
-h5py==3.6.0
+h5py
 matplotlib==3.5.2
 numpy==1.21.5
 pandas==1.4.2
@@ -16,4 +16,4 @@ torchvision==0.12.0+cu113
 tqdm==4.63.0
 transformers==4.17.0
 validators==0.18.2
-~orch==1.11.0
+