(self, x)
| 175 | self.attns = nn.ModuleList(attns) |
| 176 | |
| 177 | def forward(self, x): |
| 178 | x = self.nets[0](x) |
| 179 | for attn, net in zip(self.attns, self.nets[1:]): |
| 180 | if attn: |
| 181 | x = attn(x) |
| 182 | x = net(x) |
| 183 | return x |
| 184 | |
| 185 | |
| 186 | class UpBlock(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected