BasicBlockV2

class BasicBlockV2(channels, stride, downsample=False, in_channels=0, **kwargs)

BasicBlock V2 from “Identity Mappings in Deep Residual Networks” paper. This is used for ResNet V2 for 18, 34 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.