| 823 | 消融实验 |
| 824 | """ |
| 825 | class GCM2(nn.Module): |
| 826 | def __init__(self, in_channels, out_channels): |
| 827 | super(GCM2, self).__init__() |
| 828 | self.T1 = ETM(in_channels, out_channels) |
| 829 | self.T2 = ETM(in_channels * 2, out_channels) |
| 830 | self.T3 = ETM(in_channels * 4, out_channels) |
| 831 | self.T4 = ETM(in_channels * 8, out_channels) |
| 832 | |
| 833 | |
| 834 | def forward(self, f1, f2, f3, f4): |
| 835 | f1 = self.T1(f1) |
| 836 | f2 = self.T2(f2) |
| 837 | f3 = self.T3(f3) |
| 838 | f4 = self.T4(f4) |
| 839 | return f1,f2,f3,f4 |
| 840 | |
| 841 | class Decoder4_noEdge(nn.Module): |
| 842 | ''' |
nothing calls this directly
no outgoing calls
no test coverage detected