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:
|