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.