Build multiple branch.
(self, num_branches, block, num_blocks, num_channels)
| 113 | return Sequential(*layers) |
| 114 | |
| 115 | def _make_branches(self, num_branches, block, num_blocks, num_channels): |
| 116 | """Build multiple branch.""" |
| 117 | branches = [] |
| 118 | |
| 119 | for i in range(num_branches): |
| 120 | branches.append( |
| 121 | self._make_one_branch(i, block, num_blocks, num_channels)) |
| 122 | |
| 123 | return ModuleList(branches) |
| 124 | |
| 125 | def _make_fuse_layers(self): |
| 126 | """Build fuse layer.""" |
no test coverage detected