bitorch.models.resnet.BasicBlockV1¶
- class bitorch.models.resnet.BasicBlockV1(in_channels: int, out_channels: int, stride: int)[source]¶
BasicBlock V1 from “Deep Residual Learning for Image Recognition” paper. This is used for ResNet V1 for 18, 34 layers.
Methods
builds body and downsampling layers
forwards the input tensor x through the building block.
Attributes