I think the downsample module is different from the cifar ResNet structure from the original paper. Here seems to be a better implementation for the original ResNet structure.