implement Wave-U-Net by pytorch - View it on GitHub