search

drethage / speech-denoising-wavenet

A neural network for end-to-end speech denoising
MIT License
673 stars 165 forks source link

index 2 is out of bounds for axis 0 with size 2 #30

Closed MeiGM closed 5 years ago

MeiGM commented 5 years ago

I trained the model according to the usage instructions, and also downloaded the given data set, but the following error will occur during training. Can anyone help me? Using Theano backend. E:\anaconda\lib\site-packages\theano\gpuarray\dnn.py:184: UserWarning: Your cuDNN version is more recent than Theano. If you encounter problems, try updating Theano or downgrading cuDNN to a version >= v5 and <= v7. warnings.warn("Your cuDNN version is more recent than " Using cuDNN version 7401 on context None Mapped name None to device cuda: GeForce GTX 1060 (0000:01:00.0) Traceback (most recent call last): File "main.py", line 169, in main() File "main.py", line 165, in main inference(config, cla) File "main.py", line 108, in inference load_checkpoint=cla.load_checkpoint, print_model_summary=cla.print_model_summary) File "E:\speech-denoising-wavenet-master\models.py", line 67, in init self.model = self.setup_model(load_checkpoint, print_model_summary) File "E:\speech-denoising-wavenet-master\models.py", line 76, in setup_model model = self.build_model() File "E:\speech-denoising-wavenet-master\models.py", line 220, in build_model name='data_input_target_field_length')(data_expanded) File "E:\anaconda\lib\site-packages\keras\engine\base_layer.py", line 457, in call output = self.call(inputs, **kwargs) File "E:\speech-denoising-wavenet-master\layers.py", line 47, in call x = keras.backend.permute_dimensions(x, [0, 2, 1]) File "E:\anaconda\lib\site-packages\keras\backend\theano_backend.py", line 936, in permute_dimensions y._keras_shape = tuple(np.asarray(x._keras_shape)[list(pattern)]) IndexError: index 2 is out of bounds for axis 0 with size 2 @drethage