starts with conv layer then has `num_blocks` `ResLayer`
(self, in_channels: int, num_blocks: int, stride: int = 1)
| 57 | ) |
| 58 | |
| 59 | def make_group_layer(self, in_channels: int, num_blocks: int, stride: int = 1): |
| 60 | "starts with conv layer then has `num_blocks` `ResLayer`" |
| 61 | return [ |
| 62 | BaseConv(in_channels, in_channels * 2, ksize=3, stride=stride, act="lrelu"), |
| 63 | *[(ResLayer(in_channels * 2)) for _ in range(num_blocks)], |
| 64 | ] |
| 65 | |
| 66 | def make_spp_block(self, filters_list, in_filters): |
| 67 | m = nn.Sequential( |