(self, ca, cb)
| 131 | self.use_skip_connection = use_skip_connection |
| 132 | |
| 133 | def concat_feature(self, ca, cb): |
| 134 | if self.conv_type == 'partial': |
| 135 | ca_feature, ca_mask = ca |
| 136 | cb_feature, cb_mask = cb |
| 137 | feature_cat = torch.cat((ca_feature, cb_feature), 1) |
| 138 | # leave only the later mask |
| 139 | return feature_cat, ca_mask |
| 140 | else: |
| 141 | return torch.cat((ca, cb), 1) |
| 142 | |
| 143 | def forward(self, inp): |
| 144 | c8, c4, c2 = inp |