Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from torch.utils.data import DataLoader, ConcatDataset
|
|
9 |
import torchvision.utils as vutils
|
10 |
import random
|
11 |
|
12 |
-
LATENT_VECTOR_DIM =
|
13 |
|
14 |
class Generator_128(nn.Module):
|
15 |
def __init__(self, GPU_COUNT):
|
|
|
9 |
import torchvision.utils as vutils
|
10 |
import random
|
11 |
|
12 |
+
LATENT_VECTOR_DIM = 8 # latent vector dimension
|
13 |
|
14 |
class Generator_128(nn.Module):
|
15 |
def __init__(self, GPU_COUNT):
|