(self,z)
| 58 | self.linear3 = nn.Linear(128,256) |
| 59 | |
| 60 | def forward(self,z): |
| 61 | z = F.relu(self.linear1(z)) |
| 62 | z = F.relu(self.linear2(z)) |
| 63 | z = self.linear3(z) |
| 64 | return z |
| 65 | |
| 66 | class VariationalAutoencoder(nn.Module): |
| 67 | def __init__(self, latent_dims): |
nothing calls this directly
no outgoing calls
no test coverage detected