(self, num_branches, blocks, num_blocks, num_inchannels,
num_channels, fuse_method, multi_scale_output=True)
| 387 | |
| 388 | class HighResolutionModule(nn.Module): |
| 389 | def __init__(self, num_branches, blocks, num_blocks, num_inchannels, |
| 390 | num_channels, fuse_method, multi_scale_output=True): |
| 391 | super(HighResolutionModule, self).__init__() |
| 392 | self._check_branches( |
| 393 | num_branches, blocks, num_blocks, num_inchannels, num_channels) |
| 394 | |
| 395 | self.num_inchannels = num_inchannels |
| 396 | self.fuse_method = fuse_method |
| 397 | self.num_branches = num_branches |
| 398 | |
| 399 | self.multi_scale_output = multi_scale_output |
| 400 | |
| 401 | self.branches = self._make_branches( |
| 402 | num_branches, blocks, num_blocks, num_channels) |
| 403 | self.fuse_layers = self._make_fuse_layers() |
| 404 | self.fuse_act = nn.ReLU(False) |
| 405 | |
| 406 | def _check_branches(self, num_branches, blocks, num_blocks, num_inchannels, num_channels): |
| 407 | error_msg = '' |
nothing calls this directly
no test coverage detected