search

gabrielmittag / NISQA

NISQA - Non-Intrusive Speech Quality and TTS Naturalness Assessment
MIT License
663 stars 117 forks source link

Utilizing finetuned weights #46

Closed asosawelford closed 8 months ago

asosawelford commented 8 months ago

Hi, I've finetuned nisqa_tts.tar following the README instructions and obtained a new set of weights. However, when I try to use them for inference, I get the following error message.

python run_predict.py --mode predict_csv --pretrained_model weights/nisqa_custom.tar --csv_file /home/udesa_ubuntu/tesis/NISQA_analysis/NISQA_finetuning/test_set.csv --csv_deg stimuli --num_workers 0 --bs 10 --output_dir /home/udesa_ubuntu/tesis/NISQA_analysis/NISQA_finetuning/test

Device: cpu Model architecture: NISQA Loaded pretrained model from weights/nisqa_custom.tar Traceback (most recent call last): File "/home/udesa_ubuntu/tesis/NISQA/run_predict.py", line 42, in nisqa = nisqaModel(args) File "/home/udesa_ubuntu/tesis/NISQA/nisqa/NISQA_model.py", line 35, in init self._loadDatasets() File "/home/udesa_ubuntu/tesis/NISQA/nisqa/NISQA_model.py", line 738, in _loadDatasets self._loadDatasetsCSVpredict() File "/home/udesa_ubuntu/tesis/NISQA/nisqa/NISQA_model.py", line 818, in _loadDatasetsCSVpredict csv_con_file_path = os.path.join(self.args['data_dir'], self.args['csv_con']) File "/home/udesa_ubuntu/miniconda3/envs/nisqa/lib/python3.9/posixpath.py", line 90, in join genericpath._check_arg_types('join', a, *p) File "/home/udesa_ubuntu/miniconda3/envs/nisqa/lib/python3.9/genericpath.py", line 152, in _check_arg_types raise TypeError(f'{funcname}() argument must be str, bytes, or ' TypeError: join() argument must be str, bytes, or os.PathLike object, not 'NoneType'