Imported from https://github.com/mostafaelaraby/wavegan-pytorch