| 263 | |
| 264 | |
| 265 | class GCM_interFA(nn.Module): |
| 266 | def __init__(self, in_channels, out_channels): |
| 267 | super(GCM_interFA, self).__init__() |
| 268 | self.T1 = ETM(in_channels, out_channels) |
| 269 | self.T2 = ETM(in_channels * 2, out_channels) |
| 270 | self.T3 = ETM(in_channels * 4, out_channels) |
| 271 | self.T4 = ETM(in_channels * 8, out_channels) |
| 272 | self.interFA1 = InterFA(out_channels) |
| 273 | self.interFA2 = InterFA(out_channels) |
| 274 | self.interFA3 = InterFA(out_channels) |
| 275 | |
| 276 | def forward(self, f1, f2, f3, f4): |
| 277 | f1 = self.T1(f1) # 96,96,96 |
| 278 | f2 = self.T2(f2) # 96,48,48 |
| 279 | f3 = self.T3(f3) # 96,24,24 |
| 280 | f4 = self.T4(f4) # 96,12,12 |
| 281 | temp3, f3 = self.interFA3(f3, f4) |
| 282 | temp2, f2 = self.interFA2(f2, temp3) |
| 283 | temp1, f1 = self.interFA1(f1, temp2) |
| 284 | |
| 285 | return f1, f2, f3, f4 |
| 286 | |
| 287 | |
| 288 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected