(self)
| 39 | |
| 40 | class VAE(nn.Module): |
| 41 | def __init__(self): |
| 42 | super(VAE, self).__init__() |
| 43 | |
| 44 | self.fc1 = nn.Linear(784, 400) |
| 45 | self.fc21 = nn.Linear(400, 20) |
| 46 | self.fc22 = nn.Linear(400, 20) |
| 47 | self.fc3 = nn.Linear(20, 400) |
| 48 | self.fc4 = nn.Linear(400, 784) |
| 49 | |
| 50 | def encode(self, x): |
| 51 | h1 = F.relu(self.fc1(x)) |
nothing calls this directly
no outgoing calls
no test coverage detected