Spaces:
Runtime error
Runtime error
johnowhitaker
commited on
Commit
•
b6c4d8b
1
Parent(s):
45b6fd6
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,6 @@ import torchvision
|
|
33 |
from torchvision import transforms
|
34 |
from kornia.filters import filter2d
|
35 |
|
36 |
-
from lightweight_gan.diff_augment import DiffAugment
|
37 |
from lightweight_gan.version import __version__
|
38 |
|
39 |
from tqdm import tqdm
|
@@ -43,6 +42,13 @@ from adabelief_pytorch import AdaBelief
|
|
43 |
|
44 |
# helpers
|
45 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
46 |
def exists(val):
|
47 |
return val is not None
|
48 |
|
|
|
33 |
from torchvision import transforms
|
34 |
from kornia.filters import filter2d
|
35 |
|
|
|
36 |
from lightweight_gan.version import __version__
|
37 |
|
38 |
from tqdm import tqdm
|
|
|
42 |
|
43 |
# helpers
|
44 |
|
45 |
+
|
46 |
+
def DiffAugment(x, types=[]):
|
47 |
+
for p in types:
|
48 |
+
for f in AUGMENT_FNS[p]:
|
49 |
+
x = f(x)
|
50 |
+
return x.contiguous()
|
51 |
+
|
52 |
def exists(val):
|
53 |
return val is not None
|
54 |
|