Spaces:
Runtime error
Runtime error
Arnaudding001
commited on
Commit
•
b189b1d
1
Parent(s):
d413e42
Update encoder_encorders_psp_encoders.py
Browse files
encoder_encorders_psp_encoders.py
CHANGED
@@ -4,8 +4,8 @@ import torch.nn.functional as F
|
|
4 |
from torch import nn
|
5 |
from torch.nn import Linear, Conv2d, BatchNorm2d, PReLU, Sequential, Module
|
6 |
|
7 |
-
from
|
8 |
-
from
|
9 |
|
10 |
|
11 |
class GradualStyleBlock(Module):
|
|
|
4 |
from torch import nn
|
5 |
from torch.nn import Linear, Conv2d, BatchNorm2d, PReLU, Sequential, Module
|
6 |
|
7 |
+
from encoder_encoders_helpers import get_blocks, Flatten, bottleneck_IR, bottleneck_IR_SE
|
8 |
+
from stylegan_model import EqualLinear
|
9 |
|
10 |
|
11 |
class GradualStyleBlock(Module):
|