(self, num_branches, block, num_blocks, num_channels)
| 177 | return nn.Sequential(*layers) |
| 178 | |
| 179 | def _make_branches(self, num_branches, block, num_blocks, num_channels): |
| 180 | branches = [] |
| 181 | |
| 182 | for i in range(num_branches): |
| 183 | branches.append( |
| 184 | self._make_one_branch(i, block, num_blocks, num_channels) |
| 185 | ) |
| 186 | |
| 187 | return nn.ModuleList(branches) |
| 188 | |
| 189 | def _make_fuse_layers(self): |
| 190 | if self.num_branches == 1: |
no test coverage detected