BottleneckV1¶
-
class
BottleneckV1(channels, stride, downsample=False, in_channels=0, **kwargs)¶ Bottleneck V1 from “Deep Residual Learning for Image Recognition” paper. This is used for ResNet V1 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.