Pytorch Implementation of wavegan model to generate audio