(self, latent_avg)
| 134 | |
| 135 | class E4eEncoder(nn.Module): |
| 136 | def __init__(self, latent_avg): |
| 137 | super(E4eEncoder, self).__init__() |
| 138 | self.encoder = Encoder4Editing(50, 'ir_se', stylegan_size=1024) |
| 139 | self.latent_avg = latent_avg |
| 140 | |
| 141 | def forward(self, x): |
| 142 | codes = self.encoder(x) |
no test coverage detected