(self, x)
| 680 | self.output_dim = dim |
| 681 | |
| 682 | def forward(self, x): |
| 683 | if self.vae: |
| 684 | output = self.model(x) |
| 685 | output = output.view(x.size(0), -1) |
| 686 | output_mean = self.fc_mean(output) |
| 687 | output_var = self.fc_var(output) |
| 688 | return output_mean, output_var |
| 689 | else: |
| 690 | return self.model(x).view(x.size(0), -1) |
| 691 | |
| 692 | |
| 693 | class ContentEncoder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected