Arnaudding001 commited on
Commit
b189b1d
1 Parent(s): d413e42

Update encoder_encorders_psp_encoders.py

Browse files
Files changed (1) hide show
  1. encoder_encorders_psp_encoders.py +2 -2
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 model.encoder.encoders.helpers import get_blocks, Flatten, bottleneck_IR, bottleneck_IR_SE
8
- from model.stylegan.model import EqualLinear
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):