Update image_wgan.py

image_wgan.py

@@ -8,9 +8,9 @@ from torch.autograd import Variable
 from torch.utils.data import DataLoader
 from torchvision.utils import save_image
 
-from ml.pytorch.image_dataset import ImageDataset
-from ml.pytorch.wgan.discriminator import Discriminator
-from ml.pytorch.wgan.generator import Generator
+from image_dataset import ImageDataset
+from wgan.discriminator import Discriminator
+from wgan.generator import Generator
 
 
 class ImageWgan: