Spaces:
Runtime error
Runtime error
Update image_wgan.py
Browse files- image_wgan.py +2 -2
image_wgan.py
CHANGED
@@ -9,8 +9,8 @@ from torch.utils.data import DataLoader
|
|
9 |
from torchvision.utils import save_image
|
10 |
|
11 |
from image_dataset import ImageDataset
|
12 |
-
from
|
13 |
-
from
|
14 |
|
15 |
|
16 |
class ImageWgan:
|
|
|
9 |
from torchvision.utils import save_image
|
10 |
|
11 |
from image_dataset import ImageDataset
|
12 |
+
from discriminator import Discriminator
|
13 |
+
from generator import Generator
|
14 |
|
15 |
|
16 |
class ImageWgan:
|