Pytorch implementation of Sketch-WGAN