(self, num_branches, block, num_blocks, num_channels)
| 107 | return Sequential(*layers) |
| 108 | |
| 109 | def _make_branches(self, num_branches, block, num_blocks, num_channels): |
| 110 | branches = [] |
| 111 | |
| 112 | for i in range(num_branches): |
| 113 | branches.append( |
| 114 | self._make_one_branch(i, block, num_blocks, num_channels)) |
| 115 | |
| 116 | return ModuleList(branches) |
| 117 | |
| 118 | def _make_fuse_layers(self): |
| 119 | if self.num_branches == 1: |
no test coverage detected