search

sungwon23 / BSRNN

79 stars 13 forks source link

view_as_real is only supported for complex tensors #1

Closed SEMLLYCAT closed 1 year ago

SEMLLYCAT commented 1 year ago

Hello! I tried to change the code to dp training form and found this problem in: module.py line40. I printed the input tensor for model and found that the data form of dp multi-GPU is different from the original single GPU

sungwon23 commented 1 year ago

In module.py, The input is supposed to be complex tensor (spectogram) . If i understand your comment correctly, you can drag out the view_as_real part and change on any demand as long as that tensor is the input spectrogram.