(in_channel, depth, num_units, stride=2)
| 96 | |
| 97 | |
| 98 | def get_block(in_channel, depth, num_units, stride=2): |
| 99 | return [Bottleneck(in_channel, depth, stride)] + [Bottleneck(depth, depth, 1) for i in range(num_units - 1)] |
| 100 | |
| 101 | |
| 102 | def get_blocks(num_layers): |