RookieJunChen / FullSubNet-plus

The official PyTorch implementation of "FullSubNet+: Channel Attention FullSubNet with Complex Spectrograms for Speech Enhancement".
Apache License 2.0
242 stars 55 forks source link

AttributeError: module 'fullsubnet_plus.model.fullsubnet_plus' has no attribute 'Sub_FullSubNet_Plus' #6

Closed liziru closed 2 years ago

liziru commented 2 years ago

你好,谢谢你的工作。 我尝试训练遇到以下问题: -- Process 0 terminated with the following error: Traceback (most recent call last): File "/home/lee/Documents/software/anaconda3/envs/fsnetplus/lib/python3.8/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap fn(i, *args) File "/home/lee/Desktop/workspace/project/se/ns/tests/FullSubNet-plus/speech_enhance/tools/train.py", line 58, in entry model = initialize_module(config["model"]["path"], args=config["model"]["args"]) File "/home/lee/Desktop/workspace/project/se/ns/tests/FullSubNet-plus/speech_enhance/audio_zen/utils.py", line 91, in initialize_module class_or_function = getattr(module, class_or_function_name) AttributeError: module 'fullsubnet_plus.model.fullsubnet_plus' has no attribute 'Sub_FullSubNet_Plus'

我在工程中全局搜索没有找到这个函数或文件,请问该如何解决?

RookieJunChen commented 2 years ago

你好,不好意思,我在train.toml里的模型的地址写错了,现已纠正并更新仓库!谢谢指正!