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