(self, x: torch.Tensor, t: torch.Tensor)
| 165 | self.res2 = ResidualBlock(n_channels, n_channels, time_channels, is_noise=is_noise) |
| 166 | |
| 167 | def forward(self, x: torch.Tensor, t: torch.Tensor): |
| 168 | x = self.res1(x, t) |
| 169 | x = self.dia1(x) |
| 170 | x = self.dia2(x) |
| 171 | x = self.dia3(x) |
| 172 | x = self.dia4(x) |
| 173 | x = self.res2(x, t) |
| 174 | return x |
| 175 | |
| 176 | |
| 177 | class Upsample(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected