(self, x)
| 75 | self.layer5 = ResidualBlock(dim, 256, 256) |
| 76 | |
| 77 | def forward(self, x): |
| 78 | x = self.initial_conv(x) |
| 79 | x = self.layer1(x) |
| 80 | x = self.conv_before_res2(x) |
| 81 | x = self.layer2(x) |
| 82 | x = x.permute(0, 2, 3, 1) |
| 83 | x = self.patch_merge1(x) |
| 84 | x = x.permute(0, 3, 1, 2) |
| 85 | x = self.conv_before_res3(x) |
| 86 | x = self.layer3(x) |
| 87 | x = x.permute(0, 2, 3, 1) |
| 88 | x = self.patch_merge2(x) |
| 89 | x = x.permute(0, 3, 1, 2) |
| 90 | x = self.conv_before_res4(x) |
| 91 | x = self.layer4(x) |
| 92 | x = x.permute(0, 2, 3, 1) |
| 93 | x = self.patch_merge3(x) |
| 94 | x = x.permute(0, 3, 1, 2) |
| 95 | x = self.conv_before_res5(x) |
| 96 | x = self.layer5(x) |
| 97 | return x |
nothing calls this directly
no outgoing calls
no test coverage detected