BottleneckV2¶
-
class
BottleneckV2(channels, stride, downsample=False, in_channels=0, **kwargs)¶ Bottleneck V2 from “Identity Mappings in Deep Residual Networks” paper. This is used for ResNet V2 for 50, 101, 152 layers.
Parameters: - channels : int
Number of output channels.
- stride : int
Stride size.
- downsample : bool, default False
Whether to downsample the input.
- in_channels : int, default 0
Number of input channels. Default is 0, to infer from the graph.