From 4c274f86ce1ecd31b70a8c545a7bd61596027708 Mon Sep 17 00:00:00 2001 From: metaphysicser Date: Fri, 10 Feb 2023 10:57:51 +0800 Subject: [PATCH] fix bug: config help --- config.py | 2 +- requirements.txt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/config.py b/config.py index 67bc2b1..3f9596e 100644 --- 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 index e00ea40..79bb973 100644 --- 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 +