(self, x, skip_x, t)
| 118 | ) |
| 119 | |
| 120 | def forward(self, x, skip_x, t): |
| 121 | x = self.up(x) |
| 122 | x = torch.cat([skip_x, x], dim=1) |
| 123 | x = self.conv(x) |
| 124 | emb = self.emb_layer(t)[:, :, None, None].repeat(1, 1, x.shape[-2], x.shape[-1]) |
| 125 | return x + emb |
| 126 | |
| 127 | |
| 128 | class UNet(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected