(self, in_channels, out_channels)
| 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): |