(self, num_branches, block, num_blocks, num_channels)
| 170 | return nn.Sequential(*layers) |
| 171 | |
| 172 | def _make_branches(self, num_branches, block, num_blocks, num_channels): |
| 173 | branches = [] |
| 174 | |
| 175 | for i in range(num_branches): |
| 176 | branches.append( |
| 177 | self._make_one_branch(i, block, num_blocks, num_channels)) |
| 178 | |
| 179 | return nn.ModuleList(branches) |
| 180 | |
| 181 | def _make_fuse_layers(self): |
| 182 | if self.num_branches == 1: |
no test coverage detected